Publications

Publications by Jiaxuan Zou, including research papers and preprints on mechanistic interpretability, deep learning theory, optimization, and scaling laws.

2026

  1. Kaczmarz Linear Attention
    Kaczmarz Linear Attention
    Jiaxuan Zou, Ruifeng Ren, and Yong Liu
    2026
  2. Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizer
    Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizer
    Jinghui Yuan, Jiaxuan Zou, Shuo Wang, and 2 more authors
    2026
  3. Effective Frontiers: A Unification of Neural Scaling Laws
    Effective Frontiers: A Unification of Neural Scaling Laws
    Jiaxuan Zou, Zixuan Gong, Ye Su, and 2 more authors
    Under Review, 2026
  4. Capabilities and Fundamental Limits of Latent Chain-of-Thought
    Capabilities and Fundamental Limits of Latent Chain-of-Thought
    Jiaxuan Zou, Yaozhong Xiong, and Yong Liu
    Under Review, 2026
  5. Statistical MIA: Rethinking Membership Inference Attack for Reliable Unlearning Auditing
    Statistical MIA: Rethinking Membership Inference Attack for Reliable Unlearning Auditing
    Jialong Sun, Zeming Wei, Jiaxuan Zou, and 5 more authors
    Under Review, 2026
  6. Mamba-Driven and Feature-Fused U-Net for Automatic Seismic Horizon Interpretation
    Mamba-Driven and Feature-Fused U-Net for Automatic Seismic Horizon Interpretation
    Tian Zhang, Jiaju He, Naihao Liu, and 2 more authors
    IEEE Transactions on Geoscience and Remote Sensing, 2026

2025

  1. FreIE: Low-Frequency Spectral Bias in Neural Networks for Time-Series Tasks
    FreIE: Low-Frequency Spectral Bias in Neural Networks for Time-Series Tasks
    Jialong Sun, Xinpeng Ling, Jiaxuan Zou, and 2 more authors
    In IEEE ICDM, 2025

2024

  1. Seismic Horizon Picking Using Channel-Independent Multi-Scale UNet
    Seismic Horizon Picking Using Channel-Independent Multi-Scale UNet
    Jiaxuan Zou, Naihao Liu, Tian Zhang, and 3 more authors
    Under Review, 2024