← Search

Raghuraman Krishnamoorthi

21 accepted papers

2026

Exploring the Limits of Sub-Billion Language Model Reasoners with Open Training Recipes

ICLR 2026poster

The paradigm shift in large language models (LLMs) from instinctive responses to chain-of-thought (CoT) reasoning has fueled two prevailing assumptions: (1) reasoning capabilities only emerge in sufficiently large models, and (2) such capabilities require training on massive datasets. While the firs…

Cited by 0SourceScholar
2026

VideoAuto-R1: Video Auto Reasoning via Thinking Once, Answering Twice

CVPR 2026

Chain-of-thought (CoT) reasoning has emerged as a powerful tool for multimodal large language models on video understanding tasks. However, its necessity and advantages over direct answering remain underexplored. In this paper, we first demonstrate that for RL-trained video models, direct answering

Cited by 0SourceScholar
2026

WinQ: Accelerating Quantization-Aware Training of Large Language Models around Saddle Points

ICML 2026poster

Quantization-aware training is widely used for language model quantization in sub-4-bit precision, by training full-precision weights with gradients computed on the quantized model. The main bottleneck for this training approach is its slow convergence and plateauing of test performance, which gets …

Cited by 0SourceScholar
2026

dTRPO : Trajectory Reduction in Policy Optimization of Diffusion Large Language Models

ICML 2026poster

Diffusion Large Language Models (dLLMs) introduce a new paradigm for language generation and thus induce new challenges in aligning dLLMs for human preference. In this work, aim to optimize the dLLM generation process by developing a theoretical formulation and an efficient and effective quantificat…

Cited by 0SourceScholar
2025

Agent-as-a-Judge: Evaluate Agents with Agents

ICML 2025poster

Contemporary evaluation techniques are inadequate for agentic systems. These approaches either focus exclusively on final outcomes---ignoring the step-by-step nature of the thinking done by agentic systems---or require excessive manual labour. To address this, we introduce the **Agent-as-a-Judge** f…

2025

EdgeTAM: On-Device Track Anything Model

CVPR 2025poster

On top of Segment Anything Model (SAM), SAM 2 further extends its capability from image to video inputs through a memory bank mechanism and obtains a remarkable performance compared with previous methods, making it a foundation model for video segmentation task. In this paper, we aim at making SAM 2…

2025

LongVU: Spatiotemporal Adaptive Compression for Long Video-Language Understanding

ICML 2025poster

Multimodal Large Language Models (MLLMs) have shown promising progress in understanding and analyzing video content. However, processing long videos remains a significant challenge constrained by LLM's context size. To address this limitation, we propose \textbf{LongVU}, a spatiotemporal adaptive co…

2025

ParetoQ: Improving Scaling Laws in Extremely Low-bit LLM Quantization

NeurIPS 2025poster

The optimal bit-width for achieving the best trade-off between quantized model size and accuracy has been a subject of ongoing debate. While some advocate for 4-bit quantization, others propose that 1.58-bit offers superior results. However, the lack of a cohesive framework for different bits has le…

Cited by 0SourceScholar
2025

SpinQuant: LLM Quantization with Learned Rotations

ICLR 2025poster

Post-training quantization (PTQ) techniques applied to weights, activations, and the KV cache greatly reduce memory usage, latency, and power consumption of Large Language Models (LLMs), but may lead to large quantization errors when outliers are present. Rotating activation or weight matrices helps…

2024

Communication Efficient Distributed Training with Distributed Lion

NeurIPS 2024poster

The Lion optimizer has been a promising competitor with the AdamW for training large AI models, with advantages in memory, computation, and sample efficiency. In this paper, we introduce Distributed Lion, an innovative adaptation of Lion for distributed training environments. Leveraging the sign ope…

Cited by 4SourcePDFScholar
2024

EfficientSAM: Leveraged Masked Image Pretraining for Efficient Segment Anything

CVPR 2024highlight

Segment Anything Model (SAM) has emerged as a powerful tool for numerous vision applications. A key component that drives the impressive performance for zero-shot transfer and high versatility is a super large Transformer model trained on the extensive high-quality SA-1B dataset. While beneficial th…

2024

LLM-QAT: Data-Free Quantization Aware Training for Large Language Models

ACL 2024findings

Several post-training quantization methods have been applied to large language models (LLMs), and have been shown to perform well down to 8-bits. We find that these methods break down at lower bit precision, and investigate quantization-aware training for LLMs (LLM-QAT) to push quantization levels e…

2024

Mixture-of-Supernets: Improving Weight-Sharing Supernet Training with Architecture-Routed Mixture-of-Experts

ACL 2024findings

Weight-sharing supernets are crucial for performance estimation in cutting-edge neural architecture search (NAS) frameworks. Despite their ability to generate diverse subnetworks without retraining, the quality of these subnetworks is not guaranteed due to weight sharing. In NLP tasks like machine t…

2024

MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use Cases

ICML 2024poster

This paper addresses the growing need for efficient large language models (LLMs) on mobile devices, driven by increasing cloud costs and latency concerns. We focus on designing top-quality LLMs with fewer than a billion parameters, a practical choice for mobile deployment. Contrary to prevailing bel…

2024

Sparse Cocktail: Every Sparse Pattern Every Sparse Ratio All At Once

ICML 2024poster

Sparse Neural Networks (SNNs) have received voluminous attention for mitigating the explosion in computational costs and memory footprints of modern deep neural networks. Despite their popularity, most state-of-the-art training approaches seek to find a single high-quality sparse subnetwork with a p…

Cited by 1SourcePDFScholar
2024

TODM: Train Once Deploy Many Efficient Supernet-Based RNN-T Compression For On-Device ASR Models

ICASSP 2024accepted

Automatic Speech Recognition (ASR) models need to be optimized for specific hardware before they can be deployed on devices. This can be done by tuning the model’s hyperparameters or exploring variations in its architecture. Re-training and re-validating models after making these changes can be a re…

Cited by 0SourceScholar
2023

Binary and Ternary Natural Language Generation

ACL 2023long

Ternary and binary neural networks enable multiplication-free computation and promise multiple orders of magnitude efficiency gains over full-precision networks if implemented on specialized hardware. However, since both the parameter and the output space are highly discretized, such networks have p…

2023

Fast Point Cloud Generation With Straight Flows

CVPR 2023poster

Diffusion models have emerged as a powerful tool for point cloud generation. A key component that drives the impressive performance for generating high-quality samples from noise is iteratively denoise for thousands of steps. While beneficial, the complexity of learning steps has limited its applica…

2022

BiT: Robustly Binarized Multi-distilled Transformer

NeurIPS 2022accept

Modern pre-trained transformers have rapidly advanced the state-of-the-art in machine learning, but have also grown in parameters and computational complexity, making them increasingly difficult to deploy in resource-constrained environments. Binarization of the weights and activations of the networ…

2022

DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networks

ICML 2022spotlight

Efficient deep neural network (DNN) models equipped with compact operators (e.g., depthwise convolutions) have shown great potential in reducing DNNs’ theoretical complexity (e.g., the total number of weights/operations) while maintaining a decent model accuracy. However, existing efficient DNNs are…