Hengkang Wang

dblp:175/7774 · DBLP profile ↗
← Back
9ranked-venue papers
3as first author
8since 2021 · last 2026
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Graphics, computer vision, multimedia, augmented reality and games · 6 · 1 first-author · 5 since 2021Artificial intelligence and machine learning · 5 · 2 first-author · 5 since 2021Applied, interdisciplinary, general and emerging computing · 1 · 1 first-author · 1 since 2021
YearPublicationVenuePosition
2026 Temporal-Consistent Video Restoration with Pre-trained Diffusion Models
abstract
Video restoration (VR) aims to recover high-quality videos from degraded ones. Although recent zero-shot VR methods using pre-trained diffusion models (DMs) show good promise, they suffer from approximation errors during reverse diffusion and insufficient temporal consistency. Moreover, dealing with 3D video data, VR is inherently computationally intensive. In this paper, we advocate viewing the reverse process in DMs as a function and present a novel Maximum a Posterior (MAP) framework that directly parameterizes video frames in the seed space of DMs, eliminating approximation errors. We also introduce strategies to promote bilevel temporal consistency: semantic consistency by leveraging clustering structures in the seed space, and pixel-level consistency by progressive warping with optical flow refinements. Extensive experiments on multiple virtual reality tasks demonstrate superior visual quality and temporal consistency achieved by our method compared to the state-of-the-art.
Hengkang Wang, Huidong Liu, Chien-Chih Wang, Hongdong Li, Bryan Wang, Ju Sun
AAAI1
2024 DMPlug: A Plug-in Method for Solving Inverse Problems with Diffusion Models
abstract
Pretrained diffusion models (DMs) have recently been popularly used in solving inverse problems (IPs). The existing methods mostly interleave iterative steps in the reverse diffusion process and iterative steps to bring the iterates closer to satisfying the measurement constraint. However, such interleaving methods struggle to produce final results that look like natural objects of interest (i.e., manifold feasibility) and fit the measurement (i.e., measurement feasibility), especially for nonlinear IPs. Moreover, their capabilities to deal with noisy IPs with unknown types and levels of measurement noise are unknown. In this paper, we advocate viewing the reverse process in DMs as a function and propose a novel plug-in method for solving IPs using pretrained DMs, dubbed DMPlug. DMPlug addresses the issues of manifold feasibility and measurement feasibility in a principled manner, and also shows great potential for being robust to unknown types and levels of noise. Through extensive experiments across various IP tasks, including two linear and three nonlinear IPs, we demonstrate that DMPlug consistently outperforms state-of-the-art methods, often by large margins especially for nonlinear IPs.
Hengkang Wang, Taihui Li, Yuxiang Wan, Tiancong Chen, Ju Sun
NeurIPS1
2024 Interpretable deep learning methods for multiview learning
abstract
BACKGROUND: Technological advances have enabled the generation of unique and complementary types of data or views (e.g. genomics, proteomics, metabolomics) and opened up a new era in multiview learning research with the potential to lead to new biomedical discoveries. RESULTS: We propose iDeepViewLearn (Interpretable Deep Learning Method for Multiview Learning) to learn nonlinear relationships in data from multiple views while achieving feature selection. iDeepViewLearn combines deep learning flexibility with the statistical benefits of data and knowledge-driven feature selection, giving interpretable results. Deep neural networks are used to learn view-independent low-dimensional embedding through an optimization problem that minimizes the difference between observed and reconstructed data, while imposing a regularization penalty on the reconstructed data. The normalized Laplacian of a graph is used to model bilateral relationships between variables in each view, therefore, encouraging selection of related variables. iDeepViewLearn is tested on simulated and three real-world data for classification, clustering, and reconstruction tasks. For the classification tasks, iDeepViewLearn had competitive classification results with state-of-the-art methods in various settings. For the clustering task, we detected molecular clusters that differed in their 10-year survival rates for breast cancer. For the reconstruction task, we were able to reconstruct handwritten images using a few pixels while achieving competitive classification accuracy. The results of our real data application and simulations with small to moderate sample sizes suggest that iDeepViewLearn may be a useful method for small-sample-size problems compared to other deep learning methods for multiview learning. CONCLUSION: iDeepViewLearn is an innovative deep learning model capable of capturing nonlinear relationships between data from multiple views while achieving feature selection. It is fully open source and is freely available at https://github.com/lasandrall/iDeepViewLearn .
Hengkang Wang, Ju Sun, Sandra Safo
BMC Bioinform.1
2024 Blind Image Deblurring with Unknown Kernel Size and Substantial Noise
Zhong Zhuang, Taihui Li, Hengkang Wang, Ju Sun
Int. J. Comput. Vis.3
2023 Deep Random Projector: Accelerated Deep Image Prior
abstract
Deep image prior (DIP) has shown great promise in tackling a variety of image restoration (IR) and general visual inverse problems, needing no training data. However, the resulting optimization process is often very slow, inevitably hindering DIP's practical usage for time-sensitive scenarios. In this paper, we focus on IR, and propose two crucial modifications to DIP that help achieve substantial speedup: 1) optimizing the DIP seed while freezing randomly-initialized network weights, and 2) reducing the network depth. In addition, we reintroduce explicit priors, such as sparse gradient prior-encoded by total-variation regularization, to preserve the DIP peak performance. We evaluate the proposed method on three IR tasks, including image denoising, image super-resolution, and image inpainting, against the original DIP and variants, as well as the competing metaDIP that uses meta-learning to learn good initializers with extra data. Our method is a clear winner in obtaining competitive restoration quality in a minimal amount of time. Our code is available at https://github.com/sun-umn/Deep-Random-Projector.
Taihui Li, Hengkang Wang, Zhong Zhuang, Ju Sun
CVPR2
2023 Robust Autoencoders for Collective Corruption Removal
abstract
Robust PCA is a standard tool for learning a linear subspace in the presence of sparse corruption or rare outliers. What about robustly learning manifolds that are more realistic models for natural data, such as images? There have been several recent attempts to generalize robust PCA to manifold settings. In this paper, we propose ℓ1- and scaling-invariant ℓ1/ℓ2-robust autoencoders based on a surprisingly compact formulation built on the intuition that deep autoencoders perform manifold learning. We demonstrate on several standard image datasets that the proposed formulation significantly outperforms all previous methods in collectively removing sparse corruption, without clean images for training. Moreover, we also show that the learned manifold structures can be generalized to unseen data samples effectively.
Taihui Li, Hengkang Wang, Le Peng, Xian'e Tang, Ju Sun
ICASSP2
2023 Random Projector: Efficient Deep Image Prior
abstract
Deep image prior (DIP) has shown great promise in tackling a range of image restoration problems. However, its optimization is extremely sluggish, which inevitability hinders its practical usage when there are hard time constraints. To mitigate this issue, we propose a more compact and efficient model, dubbed random projector (RP), and freeze the convolutional layers of the neural network to prevent slow learning. We further make use of an explicit prior—total variation— to regularize the reconstructed natural images and promote pleasure-looking images. We evaluate our proposed method on different image restoration tasks such as image denoising and image inpainting, and conduct comparisons with DIP and its prevalent variants. The experimental results suggest that our proposed random projector achieves competitive restoration quality in terms of PSNR while it significantly reduces the optimization (OPT) time.
Taihui Li, Zhong Zhuang, Hengkang Wang, Ju Sun
ICASSP3
2021 Self-Validation: Early Stopping for Single-Instance Deep Generative Priors
Taihui Li, Zhong Zhuang, Hengyue Liang, Le Peng, Hengkang Wang, Ju Sun
BMVC5
2019 A 3D Cross-Hemisphere Neighborhood Difference Convnet for Chronic Stroke Lesion Segmentation
abstract
The following topics are dealt with: learning (artificial intelligence); feature extraction; convolutional neural nets; image classification; image segmentation; image representation; object detection; video signal processing; image colour analysis; neural nets.
Yanran Wang 0002, Hengkang Wang, Sophia Chen, Aggelos K. Katsaggelos, Adam Martersteck, James Higgins, Virginia B. Hill, Todd B. Parrish
ICIP2