activity
20232025
most citedMamba-UNet: UNet-Like Pure Visual Mamba for Medical Image Segmentation

88 citations · 102 across the 10 of their papers we have counts for

collaborators

12 papers

cs.CV2025

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…

cs.LG2025

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…

cs.CV2025

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…

cs.CV2025

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…

cs.CV2025

Efficient Implicit Neural Compression of Point Clouds via Learnable Activation in Latent Space

Yichi Zhang, Qianqian Yang

Implicit Neural Representations (INRs), also known as neural fields, have emerged as a powerful paradigm in deep learning, parameterizing continuous spatial fields using coordinate…

cs.LG2025

Towards Hierarchical Rectified Flow

Yichi Zhang, Yici Yan, Alex Schwing +1

We formulate a hierarchical rectified flow to model data distributions. It hierarchically couples multiple ordinary differential equations (ODEs) and defines a time-differentiable…

cs.LG2024★ 1 cited

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…

cs.CV2024★ 4 cited

Unleashing the Potential of SAM2 for Biomedical Images and Videos: A Survey

Yichi Zhang, Zhenrong Shen

The unprecedented developments in segmentation foundational models have become a dominant force in the field of computer vision, introducing a multitude of previously unexplored ca…

eess.IV2024★ 88 cited

Mamba-UNet: UNet-Like Pure Visual Mamba for Medical Image Segmentation

Ziyang Wang, Jian-Qing Zheng, Yichi Zhang +2

In recent advancements in medical image analysis, Convolutional Neural Networks (CNN) and Vision Transformers (ViT) have set significant benchmarks. While the former excels in capt…

cs.LG2024★ 2 cited

Trainable Fixed-Point Quantization for Deep Learning Acceleration on FPGAs

Dingyi Dai, Yichi Zhang, Jiahao Zhang +4

Quantization is a crucial technique for deploying deep learning models on resource-constrained devices, such as embedded FPGAs. Prior efforts mostly focus on quantizing matrix mult…

cs.CV2023★ 3 cited

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…

cond-mat.mtrl-sci2023★ 4 cited

Reconstructing microstructures from statistical descriptors using neural cellular automata

Paul Seibert, Alexander Raßloff, Yichi Zhang +4

The problem of generating microstructures of complex materials in silico has been approached from various directions including simulation, Markov, deep learning and descriptor-base…