training
moe transformer trick that reuse middle layers for training efficiency
RLT: Recurrent Looped Transformer
Diffusion Transformer
InfiLoop
기존 Looped Transformer는 마지막 recurrent state만 다음 iteration에 전달하기 때문에, 반복 횟수를 늘리면 이미 올바르게 계산한 중간 결과가 noisy update에 의해 사라지는 latent overthinking (AI Overthinking) 문제가 발생할 수 있다. InfiLoop는 이전 iteration들의 출력을 content-based weighting과 learned temporal decay로 결합하는 새로운 residual mechanism을 제안한다. 정확한 streaming recurrence를 통해 과거 결과를 누적하므로 반복 횟수가 늘어도 persistent aggregation memory는 일정하게 유지된다. 추론 depth를 무작정 늘리는 대신, 이전 iteration에서 얻은 올바른 latent state를 보존하는 mechanism이 중요하다는 연구다.

Seonglae Cho