← Search

Sanket Vaibhav Mehta

7 accepted papers

2026

Expert Routing with Synthetic Data for Domain Incremental Learning

ICML 2026poster

In many real-world settings, regulations and economic incentives permit the sharing of models but not data across institutional boundaries. In such scenarios, practitioners might hope to adapt models to new domains, without losing performance on previous domains (so-called catastrophic forgetting). …

Cited by 0SourceScholar
2025

BIG-Bench Extra Hard

ACL 2025long

Current benchmarks for large language model (LLM) reasoning predominantly focus on mathematical and coding abilities, leaving a gap in evaluating broader reasoning proficiencies. One particular exception is the BIG-Bench dataset, which has served as a crucial benchmark for evaluating the general rea…

2023

DSI++: Updating Transformer Memory with New Documents

EMNLP 2023long main

Differentiable Search Indices (DSIs) encode a corpus of documents in the parameters of a model and use the same model to map queries directly to relevant document identifiers. Despite the solid performance of DSI models, successfully deploying them in scenarios where document corpora change with tim…

Cited by 0SourceScholar
2023

Making Scalable Meta Learning Practical

NeurIPS 2023poster

Despite its flexibility to learn diverse inductive biases in machine learning programs, meta learning (i.e.,\ learning to learn) has long been recognized to suffer from poor scalability due to its tremendous compute/memory costs, training instability, and a lack of efficient distributed training sup…

2022

ExT5: Towards Extreme Multi-Task Scaling for Transfer Learning

ICLR 2022poster

Despite the recent success of multi-task learning and transfer learning for natural language processing (NLP), few works have systematically studied the effect of scaling up the number of tasks during pre-training. Towards this goal, this paper introduces ExMix (Extreme Mixture): a massive collectio…

Cited by 222SourcePDFScholar
2022

Improving Compositional Generalization with Self-Training for Data-to-Text Generation

ACL 2022long

Data-to-text generation focuses on generating fluent natural language responses from structured meaning representations (MRs). Such representations are compositional and it is costly to collect responses for all possible combinations of atomic meaning schemata, thereby necessitating few-shot general…

2022

Train Flat, Then Compress: Sharpness-Aware Minimization Learns More Compressible Models

EMNLP 2022finding

Model compression by way of parameter pruning, quantization, or distillation has recently gained popularity as an approach for reducing the computational requirements of modern deep neural network models for NLP. Inspired by prior works suggesting a connection between simpler, more generalizable mod…

Cited by 20SourcePDFScholar