- 모델이 제시하는 확률에 따른 강조 형용사 추가 ( Carrillo et al. )
- 각 형용사가 감정 강조에 미치는 영향을 나타낸 가중치 이용
- e.g. “extremely”: 95%, “very”: 75%
- 동의어 추가를 이용한 감정 레이블 반복 및 강조 ( Wang et al. )
- FAISS와 AnglE를 이용해 사용자 감정 동의어 중 프롬프트와 유사도가 높은 감정 3개를 선정하고 프롬프트 뒤에 추가
- 감정 label는 앞에, 유의어는 뒤에 붙임으로써 image-emotion alignment는 높이고 image-text alignment의 loss는 최소화한다.
- WordNet과 SenticNet에서 중복된 단어와 문맥별 단어(예: revolt, repel, four-star, first class)를 제거한 감정 동의어를 수집해 corpus를 구축
- cosine similarity를 고려하고 성능 저하를 피하는 corpus에서 상위 3개의 동의어를 선택
- AnglE를 사용해 입력으로 들어온 원본 텍스트와 감정 동의어 간의 유사도를 계산
- FAISS와 AnglE를 이용해 사용자 감정 동의어 중 프롬프트와 유사도가 높은 감정 3개를 선정하고 프롬프트 뒤에 추가
- 최종적으로 감정을 강조하는 네가지 방식을 비교함
- E1: 감정 레이블
- E2: 강조 형용사 + 감정 레이블
- E3: 감정 레이블 + 동의어 추가
- E4: 강조 형용사 + 감정 레이블 + 동의어 추가
- 감정을 고려하면서 Style keywords를 적용하는 두가지 방식을 고안하고 비교
- 사용자 감정과 일치하는 Artemis 데이터셋 캡션에서 키워드 추출
- 클러스터 생성
- LLaMA-2로 데이터 셋의 캡션에서 스타일 관련 키워드 추출
- 추출한 키워드가 포함된 텍스트 데이터를 BERT 임베딩 시킨 후 t-SNE 차원 축소 진행
- 계층적 클러스터링 진행 (자식 노드 최대 30개)
- 입력 텍스트와 가장 가까운 클러스터를 찾아, 해당 클러스터의 스타일 키워드들 중 TF (Term Frequency) 점수가 높은 상위 3개의 키워드 추출
- 클러스터 생성
- RAG 기능으로 스타일 수식어 리스트에서 키워드 선택
- MidJourney-Styles-and-Keywords-Reference에서 수집한 스타일 키워드를 color, dimensionality, style, light, perspective로 분류해 리스트로 정리
- 내용에 급격한 스타일 변화가 일어나는 수정자보다는,이미지의 시각적 요소에 변화를 주는 수정자들을 선택
- RAG: LangChain을 통해 LLaMA-2에 입력 텍스트와 감정 레이블을 query로, 스타일 수식어 리스트를 context로 제공
- vector space에서 text chunk를 나타내는 임베딩은 사전 훈련된 모델 sentence-transformers/all-MiniLM-L6-v2를 사용하여 다운로드
- 이 임베딩을 저장하고 retrieve할 vector store로는 ChromaDB를 사용
- emotion, aesthetic, context를 모두 고려하여 3개 키워드 선택
- MidJourney-Styles-and-Keywords-Reference에서 수집한 스타일 키워드를 color, dimensionality, style, light, perspective로 분류해 리스트로 정리
- 사용자 감정과 일치하는 Artemis 데이터셋 캡션에서 키워드 추출
- 최종적으로 style modifier를 적용하는 두가지 방식을 비교함
- S1: 사용자 감정과 일치하는 Artemis 데이터셋 캡션에서 키워드 추출
- S2: RAG 기능으로 스타일 수식어 리스트에서 키워드 선택
- 고정 효과: 프롬프트 엔지니어링 방법
- 랜덤 효과: 각 프롬프트의 ID
- 독립 변수: IEA, ITA, Aesthetic Score
이후 fixed 및 interaction effect에 대한 분산 분석 (ANOVA)과 사후 검정 (post-hoc test) 수행
→ 각 방법이 통계적으로 유의미한 차이를 보였음을 입증함
이미지와 감정 레이블 간의 코사인 유사도, CLIP Score 기반
이미지와 input 텍스트 간의 코사인 유사도, CLIP Score 기반
![Untitled](https://private-user-images.githubusercontent.com/77625287/339807769-3c89e34a-5074-47ea-a71b-3b5461667ff0.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjM5OTM0NTksIm5iZiI6MTcyMzk5MzE1OSwicGF0aCI6Ii83NzYyNTI4Ny8zMzk4MDc3NjktM2M4OWUzNGEtNTA3NC00N2VhLWE3MWItM2I1NDYxNjY3ZmYwLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA4MTglMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwODE4VDE0NTkxOVomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPTQzZmIwZWE3Mjk0NzMzN2ZiMTc1MjE4MTYzNGM4MDM4ZDI3MWE0ODRjODgyN2JhOWUzMmE4M2RkMTEzZTQxMzYmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.FKMFW7TE6s7lfEaEEgCuuUmi99hAyygVjs7hxKokeQw)
- Avg. score는 모든 점수를 [0, 1]로 정규화하여 계산함
- IEA와 ITA는 CLIP Score 기반으로 하기 때문에 Aesthetic Score와 스케일이 다름
- 세 가지 측면을 고려한 종합적 평가를 위해 정규화 진행
- 선형혼합회귀모델 (LMER) 피팅: E3+S2 가 IEA, Aesthetic Score 에서 가장 높은 점수를 보임
- 정규화: E3+S2 가 가장 좋은 결과를 보임
⇒ E3+S2을 Main Process 파이프라인으로 채택
: 원본 텍스트 → 감정 레이블 + 토큰화 된 텍스트 + 감정 동의어 3개 + 스타일 키워드
E1 (Baseline).ipynb 실행: main_data.csv의 utterance에 감정 레이블을 추가한다.
E2 (Intensifier).ipynb 실행: main_data.csv의 utterance에 감정 레이블과 강조 형용사를 추가한다.
E3 (Synonyms).ipynb 실행: main_data.csv의 utterance에 감정 레이블과 동의어를 추가한다.
E4 (Intensifier+Synonyms).ipynb 실행: main_data.csv의 utterance에 감정 레이블과 강조 형용사, 그리고 동의어를 추가한다.
- 랭체인_스타일키워드_레이블링.ipynb 실행: 아르테미스 데이터셋을 라마를 이용해 레이블링한다.
- 클러스터링_임베딩_차원축소.ipynb 실행: 레이블링 한 데이터를 이용해 클러스터링 진행
- 최종실험데이터_만들기.ipynb 실행: 클러스트를 만든 후 입력 데이터에 따라 스타일 키워드 추출하는 코드
- S2_get_answer.ipynb 실행: 'styleDB' 폴더를 RAG DB로 연결, LLaMA-2 답변 생성 후 'S2_answer.csv' 파일로 저장
- S2_extract_keyword_from_answer.ipynb 실행: LLaMA-2 답변에서 Style Modifiers만 추출하여 'S2_modifiers.csv' 파일로 저장
- 정규화_평균_비교.ipynb : ITA, IEA, Aethetic Score의 정규화 코드
- ImageGenerater_ClipScore_PickScore_AstheticScore.ipynb
- Overview: This project leverages the power of Stable Diffusion for image generation, enhanced with evaluation metrics including CLIP Score, PickScore, and Aesthetic Score. These metrics provide insights into the similarity between generated images and prompts, aesthetic quality, and alignment with human preferences.
- Key Components:
- Stable Diffusion Image Generation: Utilizes Stable Diffusion for generating high-quality images based on textual descriptions.
- CLIP Score: Measures the similarity between generated images and prompts, as proposed by Radford et al., 2021.
- Aesthetic Score: Evaluates the aesthetic quality of individual images following the methodology of Schuhmann et al., 2022. Implementation details are available on GitHub.
- LMER_통계.ipynb 를 실행해 1,000개 데이터들에 대한 선형혼합회귀모델 피팅 결과 표를 확인한다.
- 표의 데이터를 이용해 그래프를 그린다.