← Search

Yilei Shi

13 accepted papers

2026

AutoRegressive Generation with B-rep Holistic Token Sequence Representation

CVPR 2026

Previous representation and generation approaches for the B-rep relied on graph-based representations that disentangle geometric and topological features through decoupled computational pipelines, thereby precluding the application of sequence-based generative frameworks, such as transformer archite

Cited by 0SourcecodeScholar
2026

BrepVGAE: Variational Graph Autoencoder with Unified Latent Representation for B-rep

CVPR 2026

Due to the heterogeneity of faces and edges in B-rep, conventional graph-based representations is incapable of establishing a unified formulation for faces and edges, thereby constraining the capabilities of B-rep generative models. We propose a B-rep Variational Graph Auto Encoding (BrepVGAE), the

Cited by 0SourceScholar
2026

Dual Distillation for Few-Shot Anomaly Detection

ICLR 2026poster

Anomaly detection is a critical task in computer vision with profound implications for medical imaging, where identifying pathologies early can directly impact patient outcomes. While recent unsupervised anomaly detection approaches show promise, they require substantial normal training data and str…

Cited by 0SourcecodeScholar
2026

Learning Domain-Aware Task Prompt Representations for Multi-Domain All-in-One Image Restoration

ICLR 2026poster

Recently, significant breakthroughs have been made in all-in-one image restoration (AiOIR), which can handle multiple restoration tasks with a single model. However, existing methods typically focus on a specific image domain, such as natural scene, medical imaging, or remote sensing. In this work,…

Cited by 0SourcecodeScholar
2026

On Revisiting Entropy for Identifying Mislabeled Medical Images

ICML 2026poster

Mislabeled samples in training datasets severely degrade the performance of deep networks, as overparameterized models tend to memorize erroneous labels. We address this challenge by proposing a novel approach for mislabeled data detection that leverages training dynamics. Our method is grounded in …

Cited by 0SourceScholar
2026

PP-Brep: Few-Shot B-rep Classification with Hybrid Graph Representation

CVPR 2026

In industrial settings, classification of 3D CAD models are critical for efficient manufacturing. However, the limited availability of annotated CAD models presents an obstacle to achieving rapid adaptation in few-shot part classification scenarios. In this paper, we propose a hybrid graph represent

Cited by 0SourceScholar
2026

ProPL: Universal Semi-Supervised Ultrasound Image Segmentation via Prompt-Guided Pseudo-Labeling

AAAI 2026technical

Existing approaches for the problem of ultrasound image segmentation, whether supervised or semi-supervised, are typically specialized for specific anatomical structures or tasks, limiting their practical utility in clinical settings. In this paper, we pioneer the task of universal semi-supervised u

Cited by 0SourcePDFScholar
2025

BrepGiff: Lightweight Generation of Complex B-rep with 3D GAT Diffusion

CVPR 2025poster

Despite advancements in Computer-Aided-Design (CAD) generation, direct generation of complex Boundary Representation (B-rep) CAD models remains challenging. This difficulty arises from the parametric nature of B-rep data, complicating the encoding and generation of its geometric and topological info…

Cited by 0SourcePDFScholar
2025

LPCG: A Self-conditional Architecture for Labeled Point Cloud Generation

AAAI 2025technical

Recently, there has been considerable exploration of methods for generating 3D point clouds, which is crucial for numerous 3D vision applications. Though conditional generation methods show promising performance, it depends on the additional paired label. On the other hand, unconditional generation…

Cited by 0SourcePDFScholar
2025

MamTiff-CAD: Multi-Scale Latent Diffusion with Mamba+ for Complex Parametric Sequence

ICCV 2025poster

Parametric Computer-Aided Design (CAD) is crucial in industrial applications, yet existing approaches often struggle to generate long sequence parametric commands due to complex CAD models' geometric and topological constraints. To address this challenge, we propose MamTiff-CAD, a novel CAD parametr…

Cited by 0SourcePDFScholar
2025

Q-PART: Quasi-Periodic Adaptive Regression with Test-time Training for Pediatric Left Ventricular Ejection Fraction Regression

CVPR 2025poster

In this work, we address the challenge of adaptive pediatric Left Ventricular Ejection Fraction (LVEF) assessment. While Test-time Training (TTT) approaches show promise for this task, they suffer from two significant limitations. Existing TTT works are primarily designed for classification tasks ra…

Cited by 0SourcePDFScholar
2025

Scale-Aware Contrastive Reverse Distillation for Unsupervised Medical Anomaly Detection

ICLR 2025poster

Unsupervised anomaly detection using deep learning has garnered significant research attention due to its broad applicability, particularly in medical imaging where labeled anomalous data are scarce. While earlier approaches leverage generative models like autoencoders and generative adversarial net…

2024

Representation Enhancement-Stabilization: Reducing Bias-Variance of Domain Generalization

ECCV 2024poster

"Domain Generalization (DG) focuses on enhancing the generalization of deep learning models trained on multiple source domains to adapt to unseen target domains. This paper explores DG through the lens of bias-variance decomposition, uncovering that test errors in DG predominantly arise from cross-d…