Showing 2024 · stat.MLShow all
2 papers · 2 filters
stat.ML2024
In-context Learning for Mixture of Linear Regressions: Existence, Generalization and Training Dynamics
Yanhao Jin, Krishnakumar Balasubramanian, Lifeng Lai
We investigate the in-context learning capabilities of transformers for the -dimensional mixture of linear regression model, providing theoretical insights into their existence,…
stat.ML2024
Transformers Handle Endogeneity in In-Context Linear Regression
Haodong Liang, Krishnakumar Balasubramanian, Lifeng Lai
We explore the capability of transformers to address endogeneity in in-context linear regression. Our main finding is that transformers inherently possess a mechanism to handle end…