ParlAI는 Facebook AI Research(현재 Meta AI의 일부)가 대화 시스템 및 인공지능 모델의 훈련과 평가를 위해 개발한 오픈소스 소프트웨어 프레임워크이다. 2017년 5월에 출시되었으며, 질문 응답, 잡담, 작업 지향 대화를 포함한 광범위한 작업에서 대화형 AI 에이전트를 공유, 훈련, 테스트하기 위한 통합 플랫폼을 제공한다. 이 플랫폼은 데이터셋, 모델, 평가 지표가 통합되는 방식을 표준화하여 기계 학습 및 인공지능 연구를 촉진하도록 설계되었다.
프레임워크의 이름은 "parlay"와 "AI"의 합성어로, 여러 AI 에이전트가 상호 작용하고 서로에게서 학습할 수 있게 하는 목표를 반영한다. ParlAI는 연구자들이 신경망 모델, 사전 훈련된 트랜스포머, 외부 지식 소스와 같은 다양한 구성 요소를 단일 실험 파이프라인으로 결합할 수 있게 하는 모듈식 아키텍처를 지원한다. 이는 대화 모델의 벤치마킹과 대화형 AI 분야의 발전을 위해 학계 및 산업 연구에서 널리 채택되었다.
아키텍처 및 구성 요소
ParlAI는 세계, 에이전트, 교사라는 핵심 추상화 집합을 중심으로 구축된다. 세계는 에이전트가 상호 작용하는 환경을 정의하며, 메시지 전달과 턴 기반 진행을 처리한다. 에이전트는 AI 모델 또는 인간 참가자를 나타내며, 교사는 데이터셋에서 데이터를 제공하고 관찰과 보상을 생성하는 특수 에이전트이다. 이 설계는 단순한 단일 턴 작업부터 복잡한 다중 에이전트 대화까지 유연한 실험을 가능하게 한다.
플랫폼은 SQuAD, bAbI, Wizard of Wikipedia와 같은 대규모 통합 데이터셋 컬렉션을 포함하며, 각각은 표준화된 형식으로 래핑된다. 또한 검색 기반 및 생성 기반 접근 방식을 포함한 다양한 사전 구현 기준 모델을 제공하여 연구의 출발점으로 사용할 수 있다. 모듈성은 핵심 코드를 다시 작성하지 않고도 모델, 데이터셋, 평가 지표를 쉽게 교체할 수 있게 한다.
훈련 및 평가
ParlAI는 모델 훈련과 평가를 위한 도구를 제공하며, 여러 GPU에 걸친 분산 훈련을 지원한다. PyTorch와 같은 인기 있는 딥러닝 라이브러리와 통합되어 연구자들이 기존 딥러닝 인프라를 활용할 수 있게 한다. 프레임워크에는 혼란도, F1 점수, 정확도와 같은 내장 평가 지표와 대화형 테스트를 위한 인간 평가 인터페이스가 포함된다.
주목할 만한 기능은 두 개 이상의 모델이 서로 상호 작용하는 "자기 대결" 또는 "다중 에이전트" 시나리오를 실행할 수 있는 능력으로, 이는 창발적 행동을 연구하고 적대적 또는 협력적 에이전트를 훈련하는 데 유용하다. 플랫폼은 또한 "고정" 및 "동적" 작업을 지원하며, 후자는 즉석에서 새 예제를 생성하여 커리큘럼 학습과 지속적 적응을 가능하게 한다.
응용 및 영향
ParlAI는 생성 및 검색 접근 방식을 결합한 대규모 오픈 도메인 챗봇인 BlenderBot 모델 개발을 포함한 수많은 연구 프로젝트에서 사용되었다. 또한 Meta AI(이전 Facebook AI) 대화 연구의 기반이 되어 대규모 언어 모델 및 생성형 AI의 발전에 기여했다. 재현성과 표준화에 대한 플랫폼의 강조는 대화 시스템에 관한 학술 논문에서 일반적인 벤치마크가 되었다.
연구 외에도 ParlAI는 고객 서비스 봇, 가상 비서, 교육 도구 구축을 위한 산업 환경에서 적용되었다. 오픈소스 특성은 데이터셋 라이브러리를 확장하고 새 모델 구현을 추가한 기여자 커뮤니티를 육성하여 대화형 AI의 진화하는 환경에서 지속적인 관련성을 보장했다.
다른 플랫폼과의 관계
ParlAI는 Rasa 프레임워크 및 Microsoft Bot Framework와 같은 다른 대화 연구 플랫폼과 경쟁하고 보완하지만, 생산 중심보다는 연구 지향적이다. 관리형 AI API를 제공하는 아마존 웹 서비스 또는 구글 클라우드와 같은 상용 클라우드 서비스와 달리, ParlAI는 맞춤형 모델 개발을 위한 저수준의 유연한 환경을 제공한다. 비교 연구를 위해 오픈AI의 도구나 앤트로픽의 모델과 함께 자주 사용되지만, 특정 공급업체와 독립적이다.
플랫폼의 설계는 모델 공유를 위한 유사한 모듈식 원칙을 채택한 Hugging Face의 Transformers 라이브러리를 포함한 이후 프레임워크에 영향을 주었다. 2020년대 초반 기준으로 ParlAI는 계속 유지 관리되며, 새 데이터셋과 모델 아키텍처를 지원하기 위한 주기적 업데이트가 이루어지지만, 분야가 발전함에 따라 사용은 더 전문화된 도구로 부분적으로 이동했다.