ReBrain은 뇌 질환 진단을 위한 자기 공명 영상(MRI)을 생성하기 위해 제안된 검색 증강 확산 프레임워크입니다. 신체적 또는 임상적 제약으로 인해 MRI 촬영이 어려운 환자를 위해, 제한된 슬라이스를 가진 3차원 컴퓨터 단층 촬영(CT) 스캔으로부터 MRI를 재구성합니다. ReBrain은 Brownian Bridge Diffusion Model (BBDM)을 사용하여 MRI 슬라이스를 생성하고, 사전 데이터베이스에서 구조적 및 병리학적으로 유사한 CT 슬라이스를 검색하여 ControlNet 브랜치를 통해 MRI 생성을 안내합니다. 또한, 적절한 참조가 없는 경우를 대비하여 구면 선형 보간을 활용합니다. SynthRAD2023 및 BraTS 데이터셋에 대한 실험에서 ReBrain은 희소 조건에서 최첨단 성능을 달성했습니다.