← Search

Shihao Yang

14 accepted papers

2026

Free Energy Mixer

ICLR 2026poster

Standard attention stores keys/values losslessly but reads them via a per-head convex average, blocking channel-wise selection. We propose the Free Energy Mixer (FEM): a free-energy (log-sum-exp) read that applies a value-driven, per-channel log-linear tilt to a fast prior (e.g., from queries/keys i…

Cited by 0SourcecodeScholar
2026

HyCoRA: Hyper-Contrastive Role-Adaptive Learning for Role-Playing

AAAI 2026technical

Multi-character role-playing aims to equip models with the capability to simulate diverse roles. Existing methods either use one shared parameterized module across all roles or assign a separate parameterized module to each role. However, the role-shared module may ignore distinct traits of each rol

Cited by 0SourcePDFScholar
2026

Physics-Informed Inference Time Scaling for Solving High-Dimensional Partial Differential Equations

ICLR 2026poster

Solving high-dimensional partial differential equations (PDEs) is a critical challenge where modern data-driven solvers often lack reliability and rigorous error guarantees. We introduce Simulation-Calibrated Scientific Machine Learning (SCaSML), a framework that systematically improves pre-trained…

Cited by 0SourcecodeScholar
2026

StretchTime: Adaptive Time Series Forecasting via Symplectic Attention

ICML 2026poster

Transformer architectures have established strong baselines in time series forecasting, yet they typically rely on positional encodings that assume uniform, index-based temporal progression. However, real-world systems, from shifting financial cycles to elastic biological rhythms, frequently exhibit…

Cited by 0SourceScholar
2025

A Unified Supervised and Unsupervised Dialogue Topic Segmentation Framework Based on Utterance Pair Modeling

NAACL 2025long

The Dialogue Topic Segmentation task aims to divide a dialogue into different topic paragraphs in order to better understand the structure and content of the dialogue. Due to the short sentences, serious references and non-standard language in the dialogue, it is difficult to determine the boundarie…

Cited by 0SourcePDFScholar
2025

Inference of Whole Brain Electrophysiological Networks Through Multimodal Integration of Simultaneous Scalp and Intracranial EEG

NeurIPS 2025poster

In the past decades, brain imaging research underwent a shift from mapping tasked evoked brain regions of activations towards identifying and characterizing the dynamic brain networks of multiple coordinating brain regions. Electrophysiological signals are the direct manifestation of brain activitie…

Cited by 0SourceScholar
2025

Linear Transformers as VAR Models: Aligning Autoregressive Attention Mechanisms with Autoregressive Forecasting

ICML 2025poster

Autoregressive attention-based time series forecasting (TSF) has drawn increasing interest, with mechanisms like linear attention often outperforming vanilla attention. However, deeper Transformer architectures frequently misalign with autoregressive objectives, obscuring the underlying VAR structur…

2025

ZeroS: Zero‑Sum Linear Attention for Efficient Transformers

NeurIPS 2025spotlight

Linear attention methods offer Transformers $O(N)$ complexity but typically underperform standard softmax attention. We identify two fundamental limitations affecting these approaches: the restriction to convex combinations that only permits additive information blending, and uniform accumulated wei…

Cited by 0SourceScholar
2024

ARM: Refining Multivariate Forecasting with Adaptive Temporal-Contextual Learning

ICLR 2024poster

Long-term time series forecasting (LTSF) is important for various domains but is confronted by challenges in handling the complex temporal-contextual relationships. As multivariate input models underperforming some recent univariate counterparts, we posit that the issue lies in the inefficiency of e…

Cited by 5SourcePDFScholar
2024

CATS: Enhancing Multivariate Time Series Forecasting by Constructing Auxiliary Time Series as Exogenous Variables

ICML 2024poster

For Multivariate Time Series Forecasting (MTSF), recent deep learning applications show that univariate models frequently outperform multivariate ones. To address the deficiency in multivariate models, we introduce a method to Construct Auxiliary Time Series (CATS) that functions like a 2D temporal-…

Cited by 22SourcePDFScholar
2021

Parallel Actuation of Nanorod Swarm and Nanoparticle Swarm to Different Targets

ICRA 2021poster

After years of development, various swarms of robots have been proposed for many complicated tasks, such as forming patterns, cooperative locomotion, and adapting to different environments. However, controlling microrobotic swarms is still a challenging task owing to the lacking of integrated device…

Cited by 1SourceScholar