The Lattice of Transition Laws
확산 모델과 자기회귀(AR) 모델의 장점을 결합하는 하이브리드 모델을 위해 디코딩 스케줄을 설계하는 새로운 원리가 제시되었습니다. 이 연구는 한 모델의 디코딩 스케줄 성능을 실제 디코딩 전에 예측할 수 있는지에 초점을 맞춥니다.
연구진은 확산 모델, AR 모델, 그리고 이 둘 사이의 모델들을 하나의 오염 격자(corruption lattice) 상의 경로로 설명하고, 스케줄의 비용을 병렬 단계에서 버려지는 의존성에 정의했습니다. 이 비용은 데이터의 기하학에 의해 결정되며, 특히 마르코프 그래프 상에서 경로를 따라 의존하는 데이터의 경우, 최소 단계 수는 그래프의 트리 깊이와 같아집니다. 트리 깊이는 시퀀스 길이의 로그에 비례하고 그리드의 측면 길이에 선형적으로 비례합니다.
이러한 비용을 바탕으로 사전 학습된 가중치에서 추정한 쌍별 의존성 커널을 사용하여 디코딩 전에 스케줄의 순위를 예측합니다. 이 방법은 텍스트, 이미지, 비디오 생성에 걸쳐 다양한 지표와 벤치마크 하에서 스케줄의 순위에 대한 대부분의 예측을 검증했습니다.
결과적으로 이 연구는 향후 AR 모델, 확산 모델 및 이 둘 사이의 모든 모델에 대한 디코딩을 위한 설계 원칙을 제공합니다. 개발자들은 이 결과를 활용하여 모델의 디코딩 전략을 보다 효율적으로 설계할 수 있습니다. 코드는 https://github.com/TSUITUENYUE/The-Lattice-of-Transition-Laws에서 확인할 수 있습니다.
연구진은 확산 모델, AR 모델, 그리고 이 둘 사이의 모델들을 하나의 오염 격자(corruption lattice) 상의 경로로 설명하고, 스케줄의 비용을 병렬 단계에서 버려지는 의존성에 정의했습니다. 이 비용은 데이터의 기하학에 의해 결정되며, 특히 마르코프 그래프 상에서 경로를 따라 의존하는 데이터의 경우, 최소 단계 수는 그래프의 트리 깊이와 같아집니다. 트리 깊이는 시퀀스 길이의 로그에 비례하고 그리드의 측면 길이에 선형적으로 비례합니다.
이러한 비용을 바탕으로 사전 학습된 가중치에서 추정한 쌍별 의존성 커널을 사용하여 디코딩 전에 스케줄의 순위를 예측합니다. 이 방법은 텍스트, 이미지, 비디오 생성에 걸쳐 다양한 지표와 벤치마크 하에서 스케줄의 순위에 대한 대부분의 예측을 검증했습니다.
결과적으로 이 연구는 향후 AR 모델, 확산 모델 및 이 둘 사이의 모든 모델에 대한 디코딩을 위한 설계 원칙을 제공합니다. 개발자들은 이 결과를 활용하여 모델의 디코딩 전략을 보다 효율적으로 설계할 수 있습니다. 코드는 https://github.com/TSUITUENYUE/The-Lattice-of-Transition-Laws에서 확인할 수 있습니다.