← Search

Shrey Pandit

7 accepted papers

2026

Least-Loaded Expert Parallelism: Load Balancing An Imbalanced Mixture-of-Experts

ICML 2026poster

Mixture-of-Experts (MoE) models are typically pre-trained with explicit load-balancing constraints to ensure statistically balanced expert routing. Despite this, we observe that even well-trained MoE models exhibit significantly imbalanced routing. This behavior is arguably natural—and even desirabl…

Cited by 0SourceScholar
2025

FaithEval: Can Your Language Model Stay Faithful to Context, Even If "The Moon is Made of Marshmallows"

ICLR 2025poster

Ensuring faithfulness to context in large language models (LLMs) and retrieval-augmented generation (RAG) systems is crucial for reliable deployment in real-world applications, as incorrect or unsupported information can erode user trust. Despite advancements on standard benchmarks, faithfulness hal…

2025

MedHallu: A Comprehensive Benchmark for Detecting Medical Hallucinations in Large Language Models

EMNLP 2025

Advancements in Large Language Models (LLMs) and their increasing use in medical question-answering necessitate rigorous evaluation of their reliability. A critical challenge lies in hallucination, where models generate plausible yet factually incorrect outputs. In the medical domain, this poses ser

Cited by 0SourcePDFScholar
2024

AdaPT: A Set of Guidelines for Hyperbolic Multimodal Multilingual NLP

NAACL 2024findings

The Euclidean space is the familiar space for training neural models and performing arithmetic operations.However, many data types inherently possess complex geometries, and model training methods involve operating over their latent representations, which cannot be effectively captured in the Euclid…

2023

A Comparative Study on the Impact of Model Compression Techniques on Fairness in Language Models

ACL 2023long

Compression techniques for deep learning have become increasingly popular, particularly in settings where latency and memory constraints are imposed. Several methods, such as pruning, distillation, and quantization, have been adopted for compressing models, each providing distinct advantages. Howeve…

2022

CIAug: Equipping Interpolative Augmentation with Curriculum Learning

NAACL 2022long

Interpolative data augmentation has proven to be effective for NLP tasks. Despite its merits, the sample selection process in mixup is random, which might make it difficult for the model to generalize better and converge faster. We propose CIAug, a novel curriculum-based learning method that builds…

2022

DMix: Adaptive Distance-aware Interpolative Mixup

ACL 2022short

Interpolation-based regularisation methods such as Mixup, which generate virtual training samples, have proven to be effective for various tasks and modalities. We extend Mixup and propose DMix, an adaptive distance-aware interpolative Mixup that selects samples based on their diversity in the embed…