← Search

Kento Nishi

5 accepted papers

2025

ICLR: In-Context Learning of Representations

ICLR 2025poster

Recent work demonstrates that structured patterns in pretraining data influence how representations of different concepts are organized in a large language model’s (LLM) internals, with such representations then driving downstream abilities. Given the open-ended nature of LLMs, e.g., their ability t…

Cited by 7SourcePDFScholar
2025

Representation Shattering in Transformers: A Synthetic Study with Knowledge Editing

ICML 2025poster

Knowledge Editing (KE) algorithms alter models' weights to perform targeted updates to incorrect, outdated, or otherwise unwanted factual associations. However, recent work has shown that applying KE can adversely affect models' broader factual recall accuracy and diminish their reasoning abilities.…

Cited by 0SourcePDFScholar
2024

Joint-Task Regularization for Partially Labeled Multi-Task Learning

CVPR 2024poster

Multi-task learning has become increasingly popular in the machine learning field but its practicality is hindered by the need for large labeled datasets. Most multi-task learning methods depend on fully labeled datasets wherein each input example is accompanied by ground-truth labels for all target…

2024

Towards an Understanding of Stepwise Inference in Transformers: A Synthetic Graph Navigation Model

ICML 2024poster

Stepwise inference protocols, such as scratchpads and chain-of-thought, help language models solve complex problems by decomposing them into a sequence of simpler subproblems. To unravel the underlying mechanisms of stepwise inference we propose to study autoregressive Transformer models on a synthe…

Cited by 4SourcePDFScholar