Showing cs.LGShow all
3 papers · 1 filter
cs.LG2026
TAPIOCA: Why Task- Aware Pruning Improves OOD model Capability
Krish Sharma, Omar Naim, Soumadeep Saha +3
Recent work has promoted task-aware layer pruning as a way to improve model performance on particular tasks, as shown by TALE. In this paper, we investigate when such improvements…
cs.LG2026
TELL-TALE: Task Efficient LLMs with Task Aware Layer Elimination
Omar Naim, Krish Sharma, Niyar R Barman +1
Large Language Models (LLMs) typically come with a fixed architecture, despite growing evidence that not all layers contribute equally to every downstream task. We introduce TALE (…
cs.LG2025
Re-examining learning linear functions in context
Omar Naim, Guilhem Fouilhé, Nicholas Asher
In-context learning (ICL) has emerged as a powerful paradigm for easily adapting Large Language Models (LLMs) to various tasks. However, our understanding of how ICL works remains…