한 번 더: 강화학습 관점에서 본 신경 양자 상태(Neural Quantum States)의 재고

요약

신경 양자 상태(NQS) 최적화를 위해 강화학습의 어드밴티지 정책 경사 관점을 도입한 PWO 알고리즘을 제안합니다. PWO는 기존 Adam이나 SR 방식의 한계를 극복하여 대규모 모델에서도 안정적이고 효율적인 수렴을 보장합니다.

핵심 포인트

• NQS 최적화를 변분 에너지 최소화 기반의 정책 경사 문제로 재정의
• 신뢰 영역 알고리즘인 PWO를 통해 확장성과 이론적 보장 결합
• Ising 모델 및 스핀 시스템에서 기존 최적화 방식보다 우수한 성능 입증
• 1.5B 파라미터 RWKV-7 모델을 통해 대규모 NQS 최적화 가능성 확인