8 papers
Unsupervised Defect Detection for Surgical Instruments
Joseph Huang, Yichi Zhang, Jingxi Yu +6
Ensuring the safety of surgical instruments requires reliable detection of visual defects. However, manual inspection is prone to error, and existing automated defect detection met…
A Conditional Generative Framework for Synthetic Data Augmentation in Segmenting Thin and Elongated Structures in Biological Images
Yi Liu, Yichi Zhang
Thin and elongated filamentous structures, such as microtubules and actin filaments, often play important roles in biological systems. Segmenting these filaments in biological imag…
Robust Training of Neural Networks at Arbitrary Precision and Sparsity
Chengxi Ye, Grace Chu, Yanfeng Liu +5
The discontinuous operations inherent in quantization and sparsification introduce a long-standing obstacle to backpropagation, particularly in ultra-low precision and sparse regim…
Variational Masked Diffusion Models
Yichi Zhang, Alex Schwing, Zhizhen Zhao
Masked diffusion models have recently emerged as a flexible framework for discrete generative modeling. However, a key limitation of standard masked diffusion is its inability to e…
Diffusion Noise Feature: Accurate and Fast Generated Image Detection
Yichi Zhang, Xiaogang Xu
Generative models now produce images with such stunning realism that they can easily deceive the human eye. While this progress unlocks vast creative potential, it also presents si…
Hierarchical Rectified Flow Matching with Mini-Batch Couplings
Yichi Zhang, Yici Yan, Alex Schwing +1
Flow matching has emerged as a compelling generative modeling approach that is widely used across domains. To generate data via a flow matching model, an ordinary differential equa…