2026
AG-REPA: Causal Layer Selection for Representation Alignment in Audio Flow Matching
ICML 2026poster
REPresentation Alignment (REPA) improves the training of generative flow models by aligning intermediate hidden states with pretrained teacher features, but its effectiveness in token-conditioned audio Flow Matching critically depends on the choice of supervised layers, which is typically made heuri…