← Search

Amir Gholaminejad

4 accepted papers

2026

LoSA: Locality Aware Sparse Attention in Diffusion Language Models

ICML 2026poster

Block-wise diffusion language models (DLMs) generate multiple tokens in parallel, offering a promising alternative to autoregressive decoding. However, their inference efficiency remains bottlenecked by memory-bound attention in long-context scenarios. Naïve sparse attention is ineffective for DLMs …

Cited by 0SourceScholar
2026

Residual Context Diffusion Language Models

ICML 2026poster

Diffusion Large Language Models (dLLMs) have emerged as a promising alternative to purely autoregressive language models because they can decode multiple tokens in parallel. However, state-of-the-art block-wise dLLMs rely on a ``remasking" mechanism that decodes only the most confident tokens and di…

Cited by 0SourceScholar
2026

Reward Under Attack: Analyzing the Robustness and Hackability of Process Reward Models

ICML 2026poster

Process Reward Models (PRMs) are rapidly becoming the backbone of LLM reasoning pipelines, yet we demonstrate that state-of-the-art PRMs are systematically exploitable under optimization pressure. We introduce a three-tiered diagnostic framework that applies increasing adversarial pressure to quanti…

Cited by 0SourceScholar
2018

Shift: A Zero FLOP, Zero Parameter Alternative to Spatial Convolutions

CVPR 2018poster

Neural networks rely on convolutions to aggregate spatial information. However, spatial convolutions are expensive in terms of model size and computation, both of which grow quadratically with respect to kernel size. In this paper, we present a parameter-free, FLOP-free "shift" operation as an alter…

Cited by 494SourcePDFScholar