← Search

Wenxuan Zhong

7 accepted papers

2026

DCMM-Transformer: Degree-Corrected Mixed-Membership Attention for Medical Imaging

AAAI 2026technical

Medical images exhibit latent anatomical groupings, such as organs, tissues, and pathological regions, that standard Vision Transformers (ViTs) fail to exploit. While recent work like SBM-Transformer attempts to incorporate such structures through stochastic binary masking, they suffer from non-diff

Cited by 0SourcePDFScholar
2024

Bayesian Knowledge Distillation: A Bayesian Perspective of Distillation with Uncertainty Quantification

ICML 2024poster

Knowledge distillation (KD) has been widely used for model compression and deployment acceleration. Nonetheless, the statistical insight of the remarkable performance of KD remains elusive, and methods for evaluating the uncertainty of the distilled model/student model are lacking. To address these…

Cited by 3SourcePDFScholar
2020

Sufficient dimension reduction for classification using principal optimal transport direction

NeurIPS 2020poster

Sufficient dimension reduction is used pervasively as a supervised dimension reduction approach. Most existing sufficient dimension reduction methods are developed for data with a continuous response and may have an unsatisfactory performance for the categorical response, especially for the binary-r…

2019

Large-scale optimal transport map estimation using projection pursuit

NeurIPS 2019poster

This paper studies the estimation of large-scale optimal transport maps (OTM), which is a well known challenging problem owing to the curse of dimensionality. Existing literature approximates the large-scale OTM by a series of one-dimensional OTM problems through iterative random projection. Such me…

2019

Online Decentralized Leverage Score Sampling for Streaming Multidimensional Time Series

AISTATS 2019poster

Estimating the dependence structure of multidimensional time series data in real-time is challenging. With large volumes of streaming data, the problem becomes more difficult when the multidimensional data are collected asynchronously across distributed nodes, which motivates us to sample representa…

Cited by 25SourcePDFScholar