← Search

Mehran Shakerinava

6 accepted papers

2026

The Expressive Limits of Diagonal SSMs for State-Tracking

ICLR 2026poster

State-Space Models (SSMs) have recently been shown to achieve strong empirical performance on a variety of long-range sequence modeling tasks while remaining efficient and highly-parallelizable. However, the theoretical understanding of their expressive power remains limited. In this work, we study…

Cited by 0SourceScholar
2025

Beyond Scalar Rewards: An Axiomatic Framework for Lexicographic MDPs

NeurIPS 2025spotlight

Recent work has formalized the reward hypothesis through the lens of expected utility theory, by interpreting reward as utility. Hausner's foundational work showed that dropping the continuity axiom leads to a generalization of expected utility theory where utilities are lexicographically ordered ve…

Cited by 0SourceScholar
2024

Weight-Sharing Regularization

AISTATS 2024poster

Weight-sharing is ubiquitous in deep learning. Motivated by this, we propose a “weight-sharing regularization” penalty on the weights $w \in \mathbb{R}^d$ of a neural network, defined as $\mathcal{R}(w) = \frac{1}{d - 1}\sum_{i > j}^d |w_i - w_j|$. We study the proximal mapping of $\mathcal{R}$ and…

2022

Structuring Representations Using Group Invariants

NeurIPS 2022accept

A finite set of invariants can identify many interesting transformation groups. For example, distances, inner products and angles are preserved by Euclidean, Orthogonal and Conformal transformations, respectively. In an equivariant representation, the group invariants should remain constant on the e…

Cited by 20SourcePDFScholar