← Search

Xinwei Sun

30 accepted papers

2026

Conformal Reliability: A New Evaluation Metric for Conditional Generation

ICML 2026poster

Conditional generative models have recently achieved remarkable success in various applications. However, a suitable metric for evaluating the reliability of these models, which takes into account their inherent uncertainty, is still lacking. Existing metrics, which typically assess a single output,…

Cited by 0SourceScholar
2026

Conformalized Survival Counterfactuals Prediction for General Right-Censored Data

ICLR 2026poster

This paper aims to develop a lower prediction bound (LPB) for survival time across different treatments in the general right-censored setting. Although previous methods have utilized conformal prediction to construct the LPB, their resulting prediction sets provide only probably approximately correc…

Cited by 0SourceScholar
2026

Split Group Knockoffs: Controlling False Discovery Rate in Transformational Group Sparsity

ICML 2026poster

Controlling the false discovery rate (FDR) under complex sparsity structures remains a fundamental challenge in large language model (LLM) analysis. Motivated by multiple comparison problems in LLMs, we consider a setting in which sparsity arises at the group level after a linear transformation of m…

Cited by 0SourceScholar
2025

A Differential Inclusion Approach for Learning Heterogeneous Sparsity in Neuroimaging Analysis

AISTATS 2025poster

In voxel-based neuroimaging disease prediction, it was recently found that in addition to lesion features, there exists another type of feature called "Procedural Bias", which is introduced during preprocessing and can further improve the prediction power. However, traditional sparse learning method…

Cited by 0SourceScholar
2025

Adaptive Pruning of Pretrained Transformer via Differential Inclusions

ICLR 2025poster

Large transformers have demonstrated remarkable success, making it necessary to compress these models to reduce inference costs while preserving their performance. Current compression algorithms prune transformers at fixed compression ratios, requiring a unique pruning process for each ratio, which…

Cited by 0SourcePDFScholar
2025

Bayesian Active Learning for Bivariate Causal Discovery

ICML 2025poster

Determining the direction of relationships between variables is fundamental for understanding complex systems across scientific domains. While observational data can uncover relationships between variables, it cannot distinguish between cause and effect without experimental interventions. To effecti…

Cited by 0SourcePDFScholar
2025

Bivariate Causal Discovery with Proxy Variables: Integral Solving and Beyond

ICML 2025poster

Bivariate causal discovery is challenging when unmeasured confounders exist. To adjust for the bias, previous methods employed the proxy variable (*i.e.*, negative control outcome (NCO)) to test the treatment-outcome relationship through integral equations -- and assumed that violation of this equat…

Cited by 0SourcePDFScholar
2025

Learning Causal Alignment for Reliable Disease Diagnosis

ICLR 2025poster

Aligning the decision-making process of machine learning algorithms with that of experienced radiologists is crucial for reliable diagnosis. While existing methods have attempted to align their prediction behaviors to those of radiologists reflected in the training data, this alignment is primarily…

Cited by 0SourcePDFScholar
2025

Towards Reliable and Holistic Visual In-Context Learning Prompt Selection

NeurIPS 2025poster

Visual In-Context Learning (VICL) has emerged as a prominent approach for adapting visual foundation models to novel tasks, by effectively exploiting contextual information embedded in in-context examples, which can be formulated as a global ranking problem of potential candidates. Current VICL meth…

Cited by 0SourceScholar
2024

Causal Discovery via Conditional Independence Testing with Proxy Variables

ICML 2024poster

Distinguishing causal connections from correlations is important in many scenarios. However, the presence of unobserved variables, such as the latent confounder, can introduce bias in conditional independence testing commonly employed in constraint-based causal discovery for identifying causal relat…

2024

Doubly Robust Proximal Causal Learning for Continuous Treatments

ICLR 2024poster

Proximal causal learning is a powerful framework for identifying the causal effect under the existence of unmeasured confounders. Within this framework, the doubly robust (DR) estimator was derived and has shown its effectiveness in estimation, especially when the model assumption is violated. Howev…

2024

Exploring High-dimensional Search Space via Voronoi Graph Traversing

UAI 2024poster

Bayesian optimization (BO) is a well-established methodology for optimizing costly black-box functions. However, the sparse observations in the high-dimensional search space pose challenges in constructing reliable Gaussian Process (GP) models, which leads to blind exploration of the search space. W…

2024

LAC-Net: Linear-Fusion Attention-Guided Convolutional Network for Accurate Robotic Grasping Under the Occlusion

IROS 2024poster

This paper addresses the challenge of perceiving complete object shapes through visual perception. While prior studies have demonstrated encouraging outcomes in segmenting the visible parts of objects within a scene, amodal segmentation, in particular, has the potential to allow robots to infer the…

Cited by 1SourcecodeScholar
2023

Causal Discovery from Subsampled Time Series with Proxy Variables

NeurIPS 2023poster

Inferring causal structures from time series data is the central interest of many scientific inquiries. A major barrier to such inference is the problem of subsampling, *i.e.*, the frequency of measurement is much lower than that of causal influence. To overcome this problem, numerous methods have b…

2023

Learning Domain-Agnostic Representation for Disease Diagnosis

ICLR 2023poster

In clinical environments, image-based diagnosis is desired to achieve robustness on multi-center samples. Toward this goal, a natural way is to capture only clinically disease-related features. However, such disease-related features are often entangled with center-effect, disabling robust transferri…

Cited by 9SourcePDFScholar
2023

Out-of-distribution Representation Learning for Time Series Classification

ICLR 2023poster

Time series classification is an important problem in the real world. Due to its non-stationary property that the distribution changes over time, it remains challenging to build models for generalization to unseen distributions. In this paper, we propose to view time series classification from the d…

2023

Which Invariance Should We Transfer? A Causal Minimax Learning Approach

ICML 2023poster

A major barrier to deploying current machine learning models lies in their non-reliability to dataset shifts. To resolve this problem, most existing studies attempted to transfer stable information to unseen environments. Particularly, independent causal mechanisms-based methods proposed to remove m…

2021

Forecasting Irreversible Disease via Progression Learning

CVPR 2021poster

Forecasting Parapapillary atrophy (PPA), i.e., a symptom related to most irreversible eye diseases, provides an alarm for implementing an intervention to slow down the disease progression at early stage. A key question for this forecast is: how to fully utilize the historical data (e.g., retinal ima…

Cited by 4PDFScholar
2021

Learning Causal Semantic Representation for Out-of-Distribution Prediction

NeurIPS 2021poster

Conventional supervised learning methods, especially deep ones, are found to be sensitive to out-of-distribution (OOD) examples, largely because the learned representation mixes the semantic factor with the variation factor due to their domain-specific correlation, while only the semantic factor cau…

2021

Recovering Latent Causal Factor for Generalization to Distributional Shifts

NeurIPS 2021poster

Distributional shifts between training and target domains may degrade the prediction accuracy of learned models, mainly because these models often learn features that possess only correlation rather than causal relation with the output. Such a correlation, which is known as ``spurious correlation''…

2020

DessiLBI: Exploring Structural Sparsity of Deep Networks via Differential Inclusion Paths

ICML 2020poster

Over-parameterization is ubiquitous nowadays in training neural networks to benefit both optimization in seeking global optima and generalization in reducing prediction error. However, compressive networks are desired in many real world applications and direct training of small networks may be trapp…

2020

TCGM: An Information-Theoretic Framework for Semi-Supervised Multi-Modality Learning

ECCV 2020poster

Fusing data from multiple modalities provides more information to train machine learning systems. However, it is prohibitively expensive and time-consuming to label each modality with a large amount of data, which leads to a crucial problem of such semi-supervised multi-modal learning. Existing meth…

Cited by 25SourcePDFScholar
2019

Cascaded Generative and Discriminative Learning for Microcalcification Detection in Breast Mammograms

CVPR 2019poster

Accurate microcalcification (mC) detection is of great importance due to its high proportion in early breast cancers. Most of the previous mC detection methods belong to discriminative models, where classifiers are exploited to distinguish mCs from other backgrounds. However, it is still challenging…

Cited by 50PDFScholar
2019

iSplit LBI: Individualized Partial Ranking with Ties via Split LBI

NeurIPS 2019poster

Due to the inherent uncertainty of data, the problem of predicting partial ranking from pairwise comparison data with ties has attracted increasing interest in recent years. However, in real-world scenarios, different individuals often hold distinct preferences, thus might be misleading to merely lo…

2018

MSplit LBI: Realizing Feature Selection and Dense Estimation Simultaneously in Few-shot and Zero-shot Learning

ICML 2018oral

It is one typical and general topic of learning a good embedding model to efficiently learn the representation coefficients between two spaces/subspaces. To solve this task, $L_{1}$ regularization is widely used for the pursuit of feature selection and avoiding overfitting, and yet the sparse estima…

Cited by 23SourcePDFScholar
2016

Split LBI: An Iterative Regularization Path with Structural Sparsity

NeurIPS 2016poster

An iterative regularization path with structural sparsity is proposed in this paper based on variable splitting and the Linearized Bregman Iteration, hence called \emph{Split LBI}. Despite its simplicity, Split LBI outperforms the popular generalized Lasso in both theory and experiments. A theory of…

Cited by 26SourcePDFScholar