← Search

Weigao Sun

10 accepted papers

2026

MoM: Linear Sequence Modeling with Mixture-of-Memories

ICLR 2026poster

Linear sequence modeling methods, such as linear attention, state space modeling, and linear RNNs, offer significant efficiency improvements by reducing the complexity of training and inference. However, these methods typically compress the entire input sequence into a single fixed-size memory state…

Cited by 0SourcecodeScholar
2025

CogniBench: A Legal-inspired Framework and Dataset for Assessing Cognitive Faithfulness of Large Language Models

ACL 2025long

Faithfulness hallucinations are claims generated by a Large Language Model (LLM) not supported by contexts provided to the LLM. Lacking assessment standards, existing benchmarks focus on “factual statements” that rephrase source materials while overlooking “cognitive statements” that involve making…

2025

Improving Bilinear RNN with Closed-loop Control

NeurIPS 2025spotlight

Recent efficient sequence modeling methods, such as Gated DeltaNet, TTT, and RWKV-7, have achieved performance improvements by supervising the recurrent memory management through the Delta learning rule. Unlike previous state-space models (e.g., Mamba) and gated linear attentions (e.g., GLA), these…

Cited by 0SourceScholar
2025

Liger: Linearizing Large Language Models to Gated Recurrent Structures

ICML 2025poster

Transformers with linear recurrent modeling offer linear-time training and constant-memory inference. Despite their demonstrated efficiency and performance, pretraining such non-standard architectures from scratch remains costly and risky. The linearization of large language models (LLMs) transforms…

2025

Sequence Accumulation and Beyond: Infinite Context Length on Single GPU and Large Clusters

AAAI 2025technical

Linear sequence modeling methods, such as linear attention, state space modeling, and linear RNNs, have recently been recognized as potential alternatives to softmax attention thanks to their linear complexity and competitive performance. However, although their linear-memory advantage during traini…

Cited by 0SourcePDFScholar
2024

CO2: Efficient Distributed Training with Full Communication-Computation Overlap

ICLR 2024spotlight

The fundamental success of large language models hinges upon the efficacious implementation of large-scale distributed training techniques. Nevertheless, building a vast, high-performance cluster featuring high-speed communication interconnectivity is prohibitively costly, and accessible only to pro…

2024

MS-Net: A Multi-Path Sparse Model for Motion Prediction in Multi-Scenes

RA-L 2024

The multi-modality and stochastic characteristics of human behavior make motion prediction a highly challenging task, which is critical for autonomous driving. While deep learning approaches have demonstrated their great potential in this area, it still remains unsolved to establish a connection bet

Cited by 7SourceScholar
2024

Scaling Laws for Linear Complexity Language Models

EMNLP 2024main

The interest in linear complexity models for large language models is on the rise, although their scaling capacity remains uncertain. In this study, we present the scaling laws for linear complexity language models to establish a foundation for their scalability. Specifically, we examine the scaling…

2024

Various Lengths, Constant Speed: Efficient Language Modeling with Lightning Attention

ICML 2024poster

We present Lightning Attention, the first linear attention implementation that maintains a constant training speed for various sequence lengths under fixed memory consumption. Due to the issue with cumulative summation operations (cumsum), previous linear attention implementations cannot achieve the…

2020

pbSGD: Powered Stochastic Gradient Descent Methods for Accelerated Non-Convex Optimization

IJCAI 2020poster

We propose a novel technique for improving the stochastic gradient descent (SGD) method to train deep networks, which we term pbSGD. The proposed pbSGD method simply raises the stochastic gradient to a certain power elementwise during iterations and introduces only one additional parameter, namely,…