Skip to content

Author

Soheil Kolouri

2 papers indexed here

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 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)

Partial optimal transport compares two measures while leaving part of the mass unmatched, which is what makes it robust to outliers, occlusion, and clutter. The quantity of interest is usually the whole profile - the optimal cost at every transported cardinality - because the right amount to transport is rarely known i...

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.