In-Context Learning with Representations: Contextual Generalization of Trained Transformers - Yang, Tong; Huang, Yu; Liang, Yingbin; Chi, Yuejie (Advances in Neural Information Processing Systems, 2024)