← Search

Yash Akhauri

8 accepted papers

2026

FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusion

ICLR 2026poster

Diffusion language models offer parallel token generation and inherent bidirectionality, promising more efficient and powerful sequence modeling compared to autoregressive approaches. However, state-of-the-art diffusion models~(e.g., Dream 7B, LLaDA 8B) suffer from slow inference. While they match t…

Cited by 0SourcecodeScholar
2026

Regression Language Models for Code

ICML 2026poster

We study \textbf{code-to-metric regression}: predicting numeric outcomes of code executions, a challenging task due to the open-ended nature of programming languages. While prior methods have resorted to heavy and domain-specific feature engineering, we show that a single unified Regression Language…

Cited by 0SourceScholar
2026

xKV: Cross-Layer KV-Cache Compression via Aligned Singular Vector Extraction

ICML 2026poster

Long-context Large Language Models (LLMs) enable powerful applications but incur high memory costs due to the key–value states (KV-Cache). Recent studies attempt to share KV-Cache across layers, but these approaches either require expensive pretraining or rely on per-token cross-layer cosine similar…

Cited by 0SourceScholar
2024

ShadowLLM: Predictor-based Contextual Sparsity for Large Language Models

EMNLP 2024main

The high power consumption and latency-sensitive deployments of large language models (LLMs) have motivated efficiency techniques like quantization and sparsity. Contextual sparsity, where the sparsity pattern is input-dependent, is crucial in LLMs because the permanent removal of attention heads or…

2022

EZNAS: Evolving Zero-Cost Proxies For Neural Architecture Scoring

NeurIPS 2022accept

Neural Architecture Search (NAS) has significantly improved productivity in the design and deployment of neural networks (NN). As NAS typically evaluates multiple models by training them partially or completely, the improved productivity comes at the cost of significant carbon footprint. To alleviat…

Cited by 14SourcePDFScholar