19 citations · 20 across the 18 of their papers we have counts for
8 papers · 1 filter
Test Time Training for Supervised Causal Learning
Zizhen Deng, Jiaru Zhang, Rui Ding +5
Supervised Causal Learning (SCL) has shown promise in causal discovery by framing it as a supervised learning problem. However, it suffers from significant out-of-distribution gene…
Analytical Correction for Subsampling Bias in Drifting Models
Jiaru Zhang, Zeyun Deng, Juanwu Lu +2
Drifting models are capable one-step generative models trained to follow a drifting field. The field combines attractive and repulsive softmax-weighted centroids over the data and…
FedMomentum: Preserving LoRA Training Momentum in Federated Fine-Tuning
Peishen Yan, Yang Hua, Hao Wang +4
Federated fine-tuning of large language models (LLMs) with low-rank adaptation (LoRA) offers a communication-efficient and privacy-preserving solution for task-specific adaptation.…
PILD: Physics-Informed Learning via Diffusion
Tianyi Zeng, Tianyi Wang, Jiaru Zhang +7
Diffusion models have emerged as powerful generative tools for modeling complex data distributions, yet their purely data-driven nature limits applicability in engineering and scie…
Accelerating Inference of Discrete Autoregressive Normalizing Flows by Selective Jacobi Decoding
Jiaru Zhang, Juanwu Lu, Xiaoyu Wu +2
Discrete normalizing flows are promising generative models with advantages such as analytical log-likelihood computation and end-to-end training. However, the architectural constra…
Learning Identifiable Structures Helps Avoid Bias in DNN-based Supervised Causal Learning
Jiaru Zhang, Rui Ding, Qiang Fu +6
Causal discovery is a structured prediction task that aims to predict causal relations among variables based on their data samples. Supervised Causal Learning (SCL) is an emerging…
THOR: A Generic Energy Estimation Approach for On-Device Training
Jiaru Zhang, Zesong Wang, Hao Wang +8
Battery-powered mobile devices (e.g., smartphones, AR/VR glasses, and various IoT devices) are increasingly being used for AI training due to their growing computational power and…
Scalable Differentiable Causal Discovery in the Presence of Latent Confounders with Skeleton Posterior (Extended Version)
Pingchuan Ma, Rui Ding, Qiang Fu +4
Differentiable causal discovery has made significant advancements in the learning of directed acyclic graphs. However, its application to real-world datasets remains restricted due…