Skip to content

Author

Junming Zhang

We have 4 of 8 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

AF-Muon: An AdamW-Free Muon Optimizer for Tied-Embedding Models

Muon improves large-scale training by applying a spectral-norm steepest-descent update to matrix parameters, but practical models also contain parameter blocks that do not fit dense-matrix geometry. One important case is the tied vocabulary table, which appears in language models and other token generators and can rece...

Arash Lagzian, Paniz Halvachi, Jun-Ming Zhang et al. · 0 citations
Preprint Aug 2026

Curvature-Aware Zeroth-Order Optimization for Memory-Efficient Test-Time Adaptation

An improved ZO method is proposed to substantially boost the performance of ZO optimization based TTA, which operates by freezing pretrained weights and optimizing minimal adapter parameters via forward-only passes based gradient estimation, which can substantially reduce the memory overhead compared to BP-based method...

Jun-Ming Zhang, Shuyu Yin, Pei-Lin Liu et al. · 0 citations
#natural language process... Preprint Jul 2026

Search-G1: Grounded Search Agents via Representation-Based Intrinsic Rewards

Experiments show that Search-G1 improves the grounding--search-cost trade-off, producing shorter response-side trajectories at competitive task accuracy and periodically refits both readouts on trajectories from the latest checkpoint, allowing the reward to co-evolve with the policy.

Ruoxi Cheng, Haoxuan Ma, Hongyi Zhang et al. · 0 citations
Jul 2026

Memory Decoder at Scale: A Pretrained, Parametric Long-Term Memory

This work presents Memory Decoder at Scale, scaling memory models up to 6.9B parameters and pretraining them on 300B tokens, demonstrating that independently scaling pretrained memory offers a more parameter efficient path to improving language model performance.

Ru-Bin Wei, Jiaqi Cao, Jiarui Wang et al. · 1 citation

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