arXiv Computer Vision

PCaPaint: Prostate Cancer Inpainting by Mitigating Shortcut Learning

PCaPaint is a prostate cancer inpainting method that uses latent diffusion models (LDMs) and introduces a conditioning strategy where the condition image is filled with Gaussian noise to mitigate shortcut learning. It also proposes a new training objective that focuses on errors within the lesion region and a multi‑sequence latent design that separately compresses T2w and DWI&ADC scans to preserve their distinct frequency characteristics. Experiments show that the synthetic data generated by PCaPaint improves downstream tasks such as prostate lesion segmentation, patient‑level classification, and lesion‑level detection, outperforming recent state‑of‑the‑art LDM‑based tumor inpainting methods in both performance and image quality.

arXiv Machine Learning
Jun 18

Structural MRI Synthesis for Alzheimer's Disease via Conditional Diffusion on Anatomical Masks

arXiv:2606. 18354v1 Announce Type: cross Abstract: Recent advances in generative machine learning models have significantly improved medical imaging, offering promising solutions for data augmentation, privacy preservation, and improved model generalization.

By Muge Zhang, Muhammad Ali Khaliq, Jamal Alsakran, Byeong Kil Lee, Jeeho Ryoo
arXiv Machine Learning
Aug 24

Learning Prostate Anatomy at Test Time for Cancer Detection in Micro-Ultrasound

The paper introduces ANT, a test‑time adaptation framework that improves prostate cancer detection in micro‑ultrasound by performing a segmentation‑guided adaptation. ANT aligns a pretrained detection encoder to the target domain’s prostate anatomy using pseudo‑masks from a frozen segmentation network, thereby correcting domain‑specific feature drift while preserving cancer‑discriminative features. In a leave‑one‑center‑out evaluation, ANT raises mean AUC by 2.9% at the biopsy‑core level and 3.6% at the patient level compared to no adaptation, outperforming existing TTA baselines.

By Obed Korshie Dzikunu, Mohammad Mahdi Abootorabi, Mohamed Harmanani, Paul F. R. Wilson, Emma Willis, Ferdinand Luger, Adam Kinnaird, Brian Wodlinger, Parvin Mousavi, Purang Abolmaesumi
arXiv AI
Aug 21

MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal Prostate MRI Segmentation

arXiv:2510. 17529v3 Announce Type: replace-cross Abstract: Active Surveillance (AS) is a treatment option for managing low and intermediate-risk prostate cancer (PCa), aiming to avoid overtreatment while monitoring disease progression through serial MRI and clinical follow-up.

By Yovin Yahathugoda, Davide Prezzi, Patricia A. Gutierrez, Piyalitt Ittichaiwong, Vicky Goh, Sebastien Ourselin, Michela Antonelli
arXiv Computer Vision
Sep 22

Patch-to-Global: Random Patch Diffusion for Globally Consistent Megapixel Artifact Inpainting in Whole Slide Images

arXiv:2609.24116v1 Announce Type: new Abstract: Although deep learning has advanced Whole Slide Image (WSI) Analysis, tissue artifacts like bubbles and folds often cause silent failures by concealing...

By Hyeseong Lee, Eunsu Kim, D M Bappy, Ho Heon Kim, Youngsuk Lee, Se Young Chun, Jang-Hwan Choi, Sung Hak Lee, Sangjeong Ahn
arXiv AI
Jul 2

Controllable Diffusion-Based Lesion Inpainting for Scalable Histopathology Data Augmentation

arXiv:2601. 08127v2 Announce Type: replace-cross Abstract: Expert-annotated training data remains the critical bottleneck for AI in histopathology, particularly for rare pathologies where even dozens of cases may be unavailable.

By Mohamad Koohi-Moghadam, Mohammad-Ali Nikouei Mahani, Rex K. H. Au-Yeung, Raymond Yu O, Monalyn Marabi, Piyapharom Intarawichian, Fabian Z. X. Lean, Andrew Ferguson, Kyongtae Tyler Bae
arXiv AI
Jul 1

Learning Where to Look: A Reinforcement Learning Framework for Robust Micro-Ultrasound Prostate Cancer Detection

arXiv:2606. 30951v1 Announce Type: cross Abstract: Micro-ultrasound ($\mu$US) is a new, emerging, and promising imaging modality for prostate cancer (PCa) detection, but accurate identification of suspicious tissue remains highly dependent on clinical experience, leading to substantial inter-observer variability.

By Mohammad Mahdi Abootorabi, Sina Namazi, Armin Saadat, Lyuyang Wang, Obed Dzikunu, Paul F. R. Wilson, Zhuoxin Guo, Brian Wodlinger, Parvin Mousavi, Purang Abolmaesumi
Hugging Face Trending Papers
Jun 23

Prob-BBDM: a Probabilistic Brownian Bridge Diffusion Model for MRI sequence image-to-image translation

AI-driven image-to-image synthesis is rapidly advancing, with growing applications in medical imaging. Multi-modal image analysis plays a crucial role in optimizing examination quality, yet acquiring multiple imaging modalities in clinical settings remains resource-intensive and time-consuming, especially for 3D imaging.

arXiv AI
Sep 3

InstEditSeg: Instruction-Driven Image Editing for Polyp and Skin Lesion Segmentation

InstEditSeg is a generative framework that treats medical segmentation as an instruction-driven image editing task. Instead of producing binary masks, it renders a color-coded overlay on the original image guided by textual instructions, leveraging latent diffusion models to align with natural image distributions and reduce domain gaps. The method incorporates a DINOv3 visual encoder and a multi-scale feature pyramid fused into the diffusion U‑Net, and uses a dual‑branch classifier‑free guidance strategy to lower inference cost, achieving competitive accuracy on polyp and skin lesion datasets while improving cross‑domain generalization and multi‑lesion segmentation.

By Ziquan Liu, Zhewei Zhu, Xuyang Shi