Concept-aware Data Construction Improves In-context Learning of Language Models
The result's identifiers
Result code in IS VaVaI
<a href="https://www.isvavai.cz/riv?ss=detail&h=RIV%2F00216224%3A14330%2F24%3A00136030" target="_blank" >RIV/00216224:14330/24:00136030 - isvavai.cz</a>
Result on the web
<a href="https://openreview.net/forum?id=KwDG8YHZnP" target="_blank" >https://openreview.net/forum?id=KwDG8YHZnP</a>
DOI - Digital Object Identifier
<a href="http://dx.doi.org/10.48550/arXiv.2403.09703" target="_blank" >10.48550/arXiv.2403.09703</a>
Alternative languages
Result language
angličtina
Original language name
Concept-aware Data Construction Improves In-context Learning of Language Models
Original language description
Many recent language models (LMs) of the Transformers family are capable of in-context learning (ICL), manifested in the LMs' ability to perform a new task solely from its description in a natural language input. Previous work curating these models assumes that ICL emerges from vast over-parametrization or the scale of multi-task training, but recent theoretical work attributes ICL emergence to training data properties, creating in-context learners with small, synthetic data. Inspired by these findings, we propose Concept-aware Training (CoAT), a framework for constructing training scenarios that make it beneficial for the LM to learn to utilize the analogical reasoning concepts from demonstrations. We find that by using CoAT, pre-trained transformers can learn to better utilise new latent concepts from demonstrations and that such ability makes ICL more robust to functional deficiencies of the previous models. Finally, we show that concept-aware in-context learning improves ICL performance on a majority of new tasks compared to traditional instruction tuning, reaching performance comparable to the multitask learners using magnitudes of more training data.
Czech name
—
Czech description
—
Classification
Type
O - Miscellaneous
CEP classification
—
OECD FORD branch
10302 - Condensed matter physics (including formerly solid state physics, supercond.)
Result continuities
Project
—
Continuities
S - Specificky vyzkum na vysokych skolach
Others
Publication year
2024
Confidentiality
S - Úplné a pravdivé údaje o projektu nepodléhají ochraně podle zvláštních právních předpisů