Loop Transformer
GPT-6 Astra가 쓴 Loop Transformer (Recurrent Depth) 정리
한 줄 요약
같은 트랜스포머 블록을 여러 번 반복 통과시켜, 파라미터를 늘리지 않고도 "더 깊게 생각"하게 만드는 구조. OpenAI는 이를 recurrent depth(순환 깊이) 라 부르지만, 연구계에서 몇 년째 있던 looped transformer 아이디어의 리브랜딩이다.
어떻게 동작하나
- 입력 토큰이 트랜스포머 블록 스택(예: 22개)을 통과한다.
- 1회 통과가 끝나면 출력 hidden state를 다시 맨 앞으로 되먹임한다.
- 같은 22개 블록이 그 중간 표현을 한 번 더 처리한다. (가중치는 동일, 입력 데이터만 바뀜)
- 즉 Nanbeige4.2-3B의 경우: 22블록을 2번 → 총 44번의 "블록 적용"이지만 저장하는 가중치는 22개뿐.
왜 도움이 되나
- 파라미터 효율 — 블록을 2번 재사용하면 블록을 2배(44개)로 늘린 모델 대비 파라미터가 약 50% 적다 → 메모리 절감.
- 추론 능력 — 저장 공간(파라미터)을 늘리지 않아도 연산량(computation)을 더 쓰는 것만으로 다단계 추론이 개선됨. "저장이 아니라 계산이 핵심"이라는 가설.
헷갈리기 쉬운 것과의 차이
| 비교 대상 | 차이 |
|---|---|
| Chain-of-Thought | CoT는 보이는 추론 토큰을 생성. 루프는 내부 hidden state에서의 연산. 둘은 별개이며 함께 쓸 수 있음. |
| 그냥 레이어 추가 | 서로 다른 44블록을 쌓으면 파라미터·메모리가 2배. 루프는 절반의 파라미터로 비슷한 "계산 깊이"를 얻음(단, 학습 시 forward/backward 연산량은 동일). |
트레이드오프 / 한계
- KV 캐시는 안 줄어든다 — 패스마다 별도 캐시가 필요해 44블록 모델과 동일한 수준. (Nanbeige는 패스 간 KV 캐시 공유 시 오히려 성능 하락)
- 학습 비용은 여전히 비쌈 — 반복 적용 전부에 그래디언트가 흘러야 함.
- 최적 루프 횟수는 태스크 의존적 — Nanbeige는 2회가 가장 효율적이라 결론.
"추론을 숨긴다"는 소문에 대해
Sebastian Raschka는 루프가 chain-of-thought를 숨기는 주범은 아니라고 봄. 추론 흔적이 짧아진 건 은폐가 아니라, 모델이 더 유능해져 실수·백트래킹이 줄었기 때문 — "잘 준비된 학생일수록 연습장을 덜 쓴다".
GPT-6 Astra에 대한 추정
OpenAI 수석과학자: "Astra를 포함한 현재 프론티어 모델의 계산 그래프 깊이는 GPT-4의 2배 이내" → 2회 루핑과 일관되지만, 아키텍처 선택을 확정적으로 증명하진 않음.
참고 계보
- Universal Transformers (2018) — 토큰별로 루프 횟수를 다르게 하는 adaptive halting의 시초
- Mixture-of-Recursions (2025) — MoE식 라우터로 토큰별 루프 횟수 할당
- Latent Reasoning (2025) — 학습 중 루프 횟수를 가변(8~64)으로 두어 다양한 추론 예산에 대비
- SMELT (2026.09) — looped MoE가 동일 성능에 학습 연산 6.8~18% 절감
출처:
- Sebastian Raschka, GPT-6 Astra, Looped Transformers, and Hidden Reasoning
- arXiv: Universal Transformers / Mixture-of-Recursions / LoopMTP(2608.03624)
소스:
- GPT-6 Astra, Looped Transformers, and Hidden Reasoning — Sebastian Raschka (https://magazine.sebastianraschka.com/p/gpt-6-astra-looped-transformers-and)
- Looped Transformers and Recurrent Depth — Medium (https://levelup.gitconnected.com/looped-transformers-and-recurrent-depth-aka-what-gpt-6-astra-is-actually-doing-and-why-it-matters-c1089ac7a8c9)
- LoopMTP: A looped transformer guided by latent multi-token prediction — arXiv (https://arxiv.org/pdf/2608.03624)
댓글 0
로그인하고 댓글을 남겨보세요.
아직 댓글이 없어요.