Non-Markovian Discrete Diffusion with Causal Language Models
Created by
Haebom
저자
Yangtian Zhang, Sizhuang He, Daniel Levine, Lawrence Zhao, David Zhang, Syed A Rizvi, Emanuele Zappala, Rex Ying, David van Dijk
개요
본 논문은 이산 확산 모델의 표현력 한계를 극복하기 위해, 마르코프 가정에서 벗어나 생성 과정 전체를 고려하는 CaDDi 모델을 제안합니다. CaDDi는 순차적 추론과 시간적 추론을 통합한 비마르코프 변환기를 사용하여 과거 상태를 재방문하고 수정할 수 있도록 함으로써 오류 누적 문제를 해결합니다. 또한, 기존의 인과 언어 모델을 특수한 경우로 포함하며, 사전 훈련된 LLM 가중치를 건축학적 변경 없이 직접 재사용할 수 있습니다. 실험 결과, CaDDi는 기존 최첨단 이산 확산 모델보다 자연어 벤치마크에서 성능이 우수하며, 대규모 자기회귀 변환기와의 성능 차이를 상당히 줄였습니다.