지역 의존적 속도 조합을 통한 이산 확산의 전방 동역학 설계
Trust but Verify: Designing Region-Inhomogeneous Forward Dynamics for Discrete Diffusion
연속 확산은 전방 과정을 설계 공간으로 다루고, 이산 언어 확산은 흡수 부패를 사용하여 생성 과정에서 맥락을 완전히 신뢰한다. 본 연구에서는 흡수 및 균일 혼합 커널을 지역 의존적 속도와 결합하여 설계 공간을 재개방한다. 170M 매개변수를 가진 사전 훈련된 모델을 8,000단계 파인튜닝하는 데 5시간이 소요되며, 균일-무작위 부패에 대해 0.96의 재현율과 0.91의 정밀도로 부패한 맥락을 감지한다. 10% 부패에서 infill 토큰-F1이 3.8 포인트 개선된다.
170M 매개변수의 모델을 사용하여 부패한 맥락을 효과적으로 감지하고 개선할 수 있는 방법을 제시한다.
원문 출처
arXiv cs.LG (머신러닝)