← Search

Yimu Zhang

4 accepted papers

2026

PRAC: Principal-Random Subspace for LLM Activation Compression and Memory-Efficient Training

ICML 2026poster

Activations have become the primary memory bottleneck in large-batch LLM training. However, existing compression methods fail to exploit structure information of activations, resulting in slow convergence or limited compression. To address this, we bridge the relationship between the algorithm’s fas…

Cited by 0SourceScholar
2025

SimSort: A Data-Driven Framework for Spike Sorting by Large-Scale Electrophysiology Simulation

NeurIPS 2025poster

Spike sorting is an essential process in neural recording, which identifies and separates electrical signals from individual neurons recorded by electrodes in the brain, enabling researchers to study how specific neurons communicate and process information. Although there exist a number of spike sor…

Cited by 0SourceScholar
2024

Separation and Bias of Deep Equilibrium Models on Expressivity and Learning Dynamics

NeurIPS 2024poster

The deep equilibrium model (DEQ) generalizes the conventional feedforward neural network by fixing the same weights for each layer block and extending the number of layers to infinity. This novel model directly finds the fixed points of such a forward process as features for prediction. Despite emp…

Cited by 0SourcePDFScholar