activity
20242026
most citedTowards Identifiable Latent Additive Noise Models

1 citations · 2 across the 12 of their papers we have counts for

collaborators

13 papers

cs.LG2026

Boundary Embedding Shaping with Adaptive Contrastive Learning for Graph Structural Disentanglement

Jiaqing Chen, Zidu Yin, Yichao Cai +4

Graph neural networks (GNNs) excel at aggregating neighbor information for classification, yet their performance is hindered by graph structural entanglement, where spurious correl…

cs.LG2026

Learning Latent Dynamical Causal Processes for Single-Cell Perturbation Prediction

Wenkang Jiang, Yuhang Liu, Erdun Gao +3

Single-cell perturbation prediction aims to infer how cells respond to unseen interventions and to achieve out-of-distribution (OOD) generalization, providing a computational route…

cs.LG2026

S2Aligner: Pair-Efficient and Transferable Pre-Training for Sparse Text-Attributed Graphs

Yuhan Wang, Haopeng Zhang, Yibo Ding +6

Pre-training on text-attributed graphs (TAGs) is central to building transferable graph foundation models, where LLM-as-Aligner methods align graph and text representations through…

cs.LG2026

What Makes a Representation Good for Single-Cell Perturbation Prediction?

Wenkang Jiang, Yuhang Liu, Yichao Cai +5

Single-cell perturbation modeling is fundamental for understanding and predicting cellular responses to genetic perturbations. However, existing approaches, from causal representat…

cs.LG2026

Concept Component Analysis: A Principled Approach for Concept Extraction in LLMs

Yuhang Liu, Erdun Gao, Dong Gong +2

Developing human understandable interpretation of large language models (LLMs) becomes increasingly critical for their deployment in essential domains. Mechanistic interpretability…

cs.LG2026

The Geometric Mechanics of Contrastive Representation Learning: Alignment Potentials, Entropic Dispersion, and Cross-modal Divergence

Yichao Cai, Zhen Zhang, Yuhang Liu +1

While InfoNCE underlies modern contrastive learning, its geometric mechanisms remain under-characterized beyond the canonical alignment--uniformity decomposition. We develop a meas…

cs.CV2025

Causal Disentanglement and Cross-Modal Alignment for Enhanced Few-Shot Learning

Tianjiao Jiang, Zhen Zhang, Yuhang Liu +1

Few-shot learning (FSL) often requires effective adaptation of models using limited labeled data. However, most existing FSL methods rely on entangled representations, requiring th…

cs.LG2025

On the Value of Cross-Modal Misalignment in Multimodal Representation Learning

Yichao Cai, Yuhang Liu, Erdun Gao +4

Multimodal representation learning, exemplified by multimodal contrastive learning (MMCL) using image-text pairs, aims to learn powerful representations by aligning cues across mod…

cs.LG2025

Analytic DAG Constraints for Differentiable DAG Learning

Zhen Zhang, Ignavier Ng, Dong Gong +6

Recovering the underlying Directed Acyclic Graph (DAG) structures from observational data presents a formidable challenge, partly due to the combinatorial nature of the DAG-constra…

cs.LG2025★ 1 cited

I Predict Therefore I Am: Is Next Token Prediction Enough to Learn Human-Interpretable Concepts from Data?

Yuhang Liu, Dong Gong, Yichao Cai +6

Recent empirical evidence shows that LLM representations encode human-interpretable concepts. Nevertheless, the mechanisms by which these representations emerge remain largely unex…

cs.CE2024

InvariantStock: Learning Invariant Features for Mastering the Shifting Market

Haiyao Cao, Jinan Zou, Yuhang Liu +4

Accurately predicting stock returns is crucial for effective portfolio management. However, existing methods often overlook a fundamental issue in the market, namely, distribution…

cs.LG2024

Rethinking State Disentanglement in Causal Reinforcement Learning

Haiyao Cao, Zhen Zhang, Panpan Cai +7

One of the significant challenges in reinforcement learning (RL) when dealing with noise is estimating latent states from observations. Causality provides rigorous theoretical supp…

cs.LG2024★ 1 cited

Towards Identifiable Latent Additive Noise Models

Yuhang Liu, Zhen Zhang, Dong Gong +6

Causal representation learning (CRL) offers the promise of uncovering the underlying causal model by which observed data was generated, but the practical applicability of existing…