Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Diffusion Language Models Know the Answer Before Decoding

Created by
  • Haebom

저자

Pengxiang Li, Yefan Zhou, Dilxat Muhtar, Lu Yin, Shilin Yan, Li Shen, Yi Liang, Soroush Vosoughi, Shiwei Liu

개요

Diffusion 언어 모델(DLM)은 병렬 시퀀스 생성 및 유연한 토큰 순서를 제공하지만, 양방향 어텐션의 비용과 고품질 출력을 위한 많은 정제 단계로 인해 추론 속도가 자가 회귀 모델보다 느리다. 본 연구에서는 DLM의 조기 응답 수렴 특성을 활용하여, 최종 디코딩 단계의 절반 단계에서 정답을 내부적으로 식별할 수 있음을 발견했다. 이를 바탕으로, Prophet을 도입하여 훈련 없이 조기 커밋 디코딩을 가능하게 한다. Prophet은 상위 2개 예측 후보 간의 신뢰도 차이를 기준으로 정제 계속 여부를 동적으로 결정한다. LLaDA-8B 및 Dream-7B에 대한 실험 결과, Prophet은 디코딩 단계를 최대 3.4배 줄이면서 높은 생성 품질을 유지했다.

시사점, 한계점

DLM 추론 속도를 향상시키는 새로운 접근 방식 제시: 조기 응답 수렴 특성을 활용한 Prophet.
추가 훈련 없이 기존 DLM 구현에 통합 가능하며, 오버헤드가 적음.
디코딩 단계를 대폭 줄여 추론 속도를 향상시킴.
DLM의 조기 수렴 현상에 대한 더 깊은 이해 제공.
현재 연구는 LLaDA-8B 및 Dream-7B 모델에 대한 실험 결과만 제시함.
다른 DLM 아키텍처 및 대규모 모델에 대한 확장성 검증 필요.
조기 커밋 결정 기준인 신뢰도 차이의 최적화에 대한 추가 연구 필요.
👍