[논문 리뷰] Psychology-guided Controllable Story Generation

Yuqiang Xie, Yue Hu외 4명|arXiv (Cornell University)|2022. 10. 14.

Topic Modeling인용 수 5

한 줄 요약

이 논문은 지역적 심리상태 추적과 전반적 심리상태 기획을 통합하여 일관되고 주인공 중심의 정서적 및 욕구 기반 진행을 갖춘 스토리 생성을 가능하게 하는 심리학에 기반한 제어 가능한 스토리 생성 시스템 Pics를 제안한다. 마스로우의 욕구계층 이론과 플루치크의 감정 휠을 활용해 단기적인 정서 변화와 장기적인 심리적 궤적을 모델링함으로써 Pics는 자동 평가 및 인간 평가에서 베이스라인을 능가하며, 생성된 서사에서 더 높은 일관성과 심리적 일관성을 보여준다.

ABSTRACT

Controllable story generation is a challenging task in the field of NLP, which has attracted increasing research interest in recent years. However, most existing works generate a whole story conditioned on the appointed keywords or emotions, ignoring the psychological changes of the protagonist. Inspired by psychology theories, we introduce global psychological state chains, which include the needs and emotions of the protagonists, to help a story generation system create more controllable and well-planned stories. In this paper, we propose a Psychology-guIded Controllable Story Generation System (PICS) to generate stories that adhere to the given leading context and desired psychological state chains for the protagonist. Specifically, psychological state trackers are employed to memorize the protagonist's local psychological states to capture their inner temporal relationships. In addition, psychological state planners are adopted to gain the protagonist's global psychological states for story planning. Eventually, a psychology controller is designed to integrate the local and global psychological states into the story context representation for composing psychology-guided stories. Automatic and manual evaluations demonstrate that PICS outperforms baselines, and each part of PICS shows effectiveness for writing stories with more consistent psychological changes.

연구 동기 및 목표

• 기존의 제어 가능한 스토리 생성 모델이 주인공의 동적인 심리적 변화를 모델링하지 못하는 한계를 해결하기 위해.
• 단기적(지역적) 및 장기적(전반적) 심리상태 체인(욕구 및 감정 포함)을 스토리 생성에 통합하여 더 높은 서사 제어를 가능하게 하기 위해.
• 심리상태 추적 및 기획을 통합한 유일한 트랜스포머 기반 아키텍처를 통해 스토리의 일관성과 심리적 일관성을 향상시키기 위해.
• 심리상태 추적, 기획, 제어기 구성 요소의 효과성을 평가하여 원하는 심리적 궤적에 부합하는 스토리 생성 여부를 확인하기 위해.

제안 방법

• 심리상태 추적기는 각 스토리 단계에서 주인공의 지역적 심리상태(욕구 및 감정)를 기억하고 업데이트하여 시간적 관계를 포착한다.
• 심리상태 기획기는 원하는 서사 궤적을 바탕으로 주인공의 전반적 심리상태 체인(욕구 및 감정의 순서)을 생성함으로써 장기적인 스토리 기획을 가능하게 한다.
• 심리학 제어기는 지역적 및 전반적 심리상태를 스토리 컨텍스트 표현에 통합하여 생성 모델이 심리적으로 일관된 출력을 생성하도록 이끈다.
• 시스템은 트랜스포머 기반 아키텍처에 기반하며, 심리상태 인식 조건부를 갖춘 사전학습된 언어 모델(예: BART)을 미세조정한다.
• 모델은 스토리 공감각 데이터셋에서 훈련 및 평가되며, 주어진 컨텍스트 및 원하는 심리상태 체인을 포함한 제어 입력을 사용한다.
• 역사적 추론 분석 및 인간 평가를 통해 심리상태 구성 요소가 스토리 품질에 미치는 영향을 분석한다.

실험 결과

연구 질문

• RQ1 지역적 및 전반적 심리상태를 함께 모델링하면 스토리 생성의 일관성과 제어 가능성 향상에 기여하는가?
• RQ2 심리상태 추적기 및 기획기는 어떻게 일관된 정서적 및 욕구 기반 진행을 갖춘 스토리를 생성하는 데 기여하는가?
• RQ3 키워드나 감정에만 기반한 베이스라인 모델과 비교해 심리상태 정보를 통합할 경우 서사 품질이 얼마나 향상되는가?
• RQ4 심리학에 기반한 스토리 생성에서 가장 흔한 오류는 무엇이며, 기존의 생성 모델과는 어떻게 다를까?

주요 결과

• Pics는 GPT-2 및 BART와 같은 강력한 베이스라인 모델보다 자동 평가 및 인간 평가 모두에서 뚜렷이 승리하며, 특히 심리적 일관성과 서사 일관성에서 뛰어난 성능을 보였다.
• 심리상태 추적기 및 기획기 구성 요소는 각각 스토리 품질 향상에 기여하며, 추론 분석 결과에서 둘 중 하나를 제거할 경우 성능이 유의미하게 저하됨을 확인했다.
• 심리학 제어기는 지역적 및 전반적 심리상태를 생성 과정에 효과적으로 통합하여 원하는 심리상태 체인과 밀도 높게 일치하는 스토리를 생성했다.
• 오류 분석 결과 반복, 상충하는 심리상태, 모호한 심리상태가 주요 실패 유형으로 나타났으며, 반복은 전체 성능 향상에도 불구하고 Pics에서 특히 빈번하게 발생했다.
• 인간 평가 결과 Pics가 GPT-2와의 64개의 페어 비교 중 56개, BART와의 100개의 페어 비교 중 100개에서 선호되었으며, 이는 인식된 품질에서의 뚜렷한 열세를 보여주었다.
• 모델는 심리상태(예: 굶주림이 행동으로 이어짐, 분노가 갈등으로 이어짐)가 사건과 인과적으로 연결된 스토리를 성공적으로 생성하여 더 나은 서사 논리성을 입증했다.