works on

From the 1 of 17 linked papers with an AI index.

activity
20242026
collaborators
Showing cs.LGShow all

7 papers · 1 filter

cs.LG2026

PRISM Edit: One Vector for All Temporal Answers

Chen Huang, Qi Zheng, Ruiqin Zheng +2

The paper proposes PRISM Edit, a method that updates large language models to handle changing temporal facts by learning a single representation that can be modulated for different…

cs.LG2026

Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention

Jing Huang, Daniel Wurgaft, Rachit Bansal +6

Larger models learn tasks smaller models do not. What drives this phenomenon? We develop a simple phenomenological argument that power-law scaling already suggests that a larger mo…

cs.LG2026

PreFT: Prefill-only finetuning for efficient inference

Andrew Lanpouthakoun, Aryaman Arora, Zhengxuan Wu +4

Large language models can now be personalised efficiently at scale using parameter efficient finetuning methods (PEFTs), but serving user-specific PEFTs harms throughput, even with…

cs.LG2025

Internal Causal Mechanisms Robustly Predict Language Model Out-of-Distribution Behaviors

Jing Huang, Junyi Tao, Thomas Icard +2

Interpretability research now offers a variety of techniques for identifying abstract internal mechanisms in neural networks. Can such techniques be used to predict how models will…

cs.LG2025

Blackbox Model Provenance via Palimpsestic Membership Inference

Rohith Kuditipudi, Jing Huang, Sally Zhu +3

Suppose Alice trains an open-weight language model and Bob uses a blackbox derivative of Alice's model to produce text. Can Alice prove that Bob is using her model, either by query…

cs.LG2024

MoEUT: Mixture-of-Experts Universal Transformers

Róbert Csordás, Kazuki Irie, Jürgen Schmidhuber +2

Previous work on Universal Transformers (UTs) has demonstrated the importance of parameter sharing across layers. By allowing recurrence in depth, UTs have advantages over standard…