6 papers
MediEncoder: Nonlinear Representation Learning for High-Dimensional Causal Mediation Analysis
Shi Bo, Debarghya Mukherjee, AmirEmad Ghassami
Causal mediation analysis decomposes a treatment effect into indirect pathways through mediators and direct pathways not operating through them. Modern biomedical studies often inv…
Towards General Preference Alignment: Diffusion Models at Nash Equilibrium
Jiaming Hu, Jiamu Bai, Haoyu Wang +2
Reinforcement learning from human feedback (RLHF) has been popular for aligning text-to-image (T2I) diffusion models with human preferences. As a mainstream branch of RLHF, Direct…
CCFC: Core & Core-Full-Core Dual-Track Defense for LLM Jailbreak Protection
Jiaming Hu, Haoyu Wang, Debarghya Mukherjee +1
Jailbreak attacks pose a serious challenge to the safe deployment of large language models (LLMs). We introduce CCFC (Core & Core-Full-Core), a dual-track, prompt-level defense fra…
Debiased Causal Mediation Analysis in Ultra-High-Dimensional Settings in the Presence of Interaction Effects
Shi Bo, AmirEmad Ghassami, Debarghya Mukherjee
Mediation analysis is a crucial tool for uncovering the mechanisms through which a treatment affects an outcome, providing deeper causal insights and guiding effective intervention…
DRO-Augment Framework: Robustness by Synergizing Wasserstein Distributionally Robust Optimization and Data Augmentation
Jiaming Hu, Debarghya Mukherjee, Ioannis Ch. Paschalidis
In many real-world applications, ensuring the robustness and stability of deep neural networks (DNNs) is crucial, particularly for image classification tasks that encounter various…
Transfer Learning Under High-Dimensional Network Convolutional Regression Model
Liyuan Wang, Jiachen Chen, Kathryn L. Lunetta +3
Transfer learning enhances model performance by utilizing knowledge from related domains, particularly when labeled data is scarce. While existing research addresses transfer learn…