Skip to content

Author

Mohsen Imani

5 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.

Preprint Jul 2026

ExaGEMM: Exploration Framework for CPU-Driven ML Inference via Associative In-Register Computing for Low-Bit GEMM

Low-bit GEMM is increasingly central to efficient ML inference, yet very-low-bit execution remains a poor fit for conventional CPUs. Practical deployment spans fragmented regimes-from 1/2/4-bit weights to varying activation precision-whose feasibility, reuse opportunity, and support cost differ under fixed SIMD and register-file budgets, making lightweight CPU support selection a first-class design problem. We present ExaGEMM, a workload-aware codesign and exploration framework for CPU-native low-bit GEMM via register-resident LUT execution. The key insight is that existing SIMD datapaths already cover table generation and accumulation; the only new hardware is an in-register select/feed mechanism with explicitly modeled cost. ExaGEMM co-explores parameterized kernels and lightweight SIMD ISA support using analytical models of register feasibility, compute cost, memory traffic, and hardware overhead, pruning the candidate space by 99.2% before simulation. It then identifies non-dominated support points and generates ISA specs, gem5 patches, and GEMM kernels for validation. Across representative ML models and CPU targets, ExaGEMM improves latency by 13.29x over software-only baselines, while showing that workload-aware frontier selection is especially important for mixed-precision LLM workloads.

Hyunwoo Oh, Suyeon Jang, Hanning Chen et al. · 0 citations
Preprint Aug 2026

Trident : How to Break Deep Reinforcement Learning Cyber Defenses (Agentic)

Empirical evaluations reveal a fundamental brittleness in existing defenses: with a single trainable 7B planner, Trident reduces blue agent defensive performance by an average of 522% compared to static red agent baselines while autonomously discovering emergent behaviors such as decoy avoidance and adaptive state prioritization that static heuristics entirely fail to uncover.

Ryozo Masukawa, Ian Bryant, Armita Kazeminajafabadi et al. · 0 citations
#small language model Preprint Aug 2026

Asymmetric Capacity Allocation in Self-Refinement Pipelines

It is concluded that larger generators and refiners generally improve the pipeline, whereas an undersized refiner can even harm performance, and that model capacity should not be allocated uniformly across self-refinement pipelines.

Zhuoyi Yang, Ian G. Harris, Salar Hashemitaheri et al. · 0 citations
Preprint Jul 2026

PolyQ: Codesigning End-to-End Quantization Framework for Scalable Edge CPU LLM Inference

PolyQ, a CPU-oriented compiler/quantization co-design for activation-aware channel-wise bit allocation under a user-specified average-bit budget, shows that fractional-bit CPU deployment is practical, predictable, and energy-efficient across diverse edge targets.

Hyunwoo Oh, Suyeon Jang, Hanning Chen et al. · 0 citations
#artificial intelligence Preprint Aug 2026

Vector Symbolic Policy Gradient

Vector-Symbolic Policy Gradient (VSPG), a discrete-action actor that represents each action by a unit-norm hypervector and scores it by similarity to the encoded state, connects VSA action memories, log-linear policy gradients, and kernel policy search while providing a quantitative robustness guarantee.

Ryozo Masukawa, Sanggeon Yun, Sungheon Jeong et al. · 0 citations