Skip to content

Author

Soheil Kolouri

We have 4 of 146 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#machine learning Preprint Oct 2026

OVAL: Output-Aware Local Page Bases for KV Cache Retrieval

Long context inference with large language models becomes increasingly expensive as attention must operate over an ever growing KV cache. Page sparse attention reduces this cost by representing each KV page compactly and retrieving only a subset for each query. Existing retrieval methods are designed to estimate attent...

Ashkan Shahbazi, Chayne Thrash, Soheil Kolouri · 0 citations
#machine learning Preprint Sep 2026

BASE: Batch-Aware Selection of Experts Using Predicted Removal Error for Efficient MoE Decoding

Large language models are increasingly expensive to serve. In large-scale serving systems, autoregressive decoding is often bottlenecked by transferring model weights from accelerator high-bandwidth memory into on-chip SRAM. Mixture-of-experts (MoE) models reduce computation by activating only a small subset of experts...

Ali Abbasi, Justin Shi, Soheil Kolouri · 0 citations
#machine learning Preprint Sep 2026

Output-aware Residual Stream Pruning for Large Language Models

Residual stream pruning methods reduce inference cost by shrinking the model's hidden dimension, but existing approaches typically choose these dimensions by minimizing activation reconstruction error. This criterion implicitly treats all perturbation directions as equally important, ignoring the sensitivity of downstr...

Chayne Thrash, Ke Chen, Soheil Kolouri · 0 citations
Preprint Aug 2026

Partial Optimal Transport on the Circle for All Transported Masses in O(N log N)

PAWC is proposed: an exact $O(N\log N)$ time, $O(N)$ memory algorithm returning all $K+1$ costs, nested active sets and plans in one run, together with a single gap that is simultaneously optimal for every cardinality.

Soheil Kolouri · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.