Poster (After Call) FZJ-2025-02903

http://join2-wiki.gsi.de/foswiki/pub/Main/Artwork/join2_logo100x88.png
Step by Step: Towards a gapless 1 micron BigBrain with Diffusion Models

 ;  ;  ;

2025

Helmholtz AI Conference 2025, HAICON25, KarlsruheKarlsruhe, Germany, 3 Jun 2025 - 5 Jun 20252025-06-032025-06-05

Abstract: Advances in microscopic imaging and high-performance computing allow analyzing the complex cellular structure of the human brain in great detail. This progress has greatly aided in brain mapping and cell segmentation, and the development of automated analysis methods. However, histological image data can contain data gaps due to inevitable processing artifacts, which, despite careful precautions, may arise during histological lab work, such as missing sections, tissue tears, or inconsistent staining.To address this issue, we presented a convolutional neural network model that reconstructs corrupted data from surrounding tissue, while preserving precise cellular distributions. Our approach uses a denoising diffusion probabilistic model trained on light-microscopy scans of cell-body stained histological sections. We extended this model with the RePaint method to impute corrupted image data. We evaluate its performance with established deep learning models trained on the same type of histological data.A key challenge of our initial model was its difficulty in accurately reconstructing tissue boundaries and larger anatomical structures such as blood vessels. We address these challenges by an enhanced diffusion-based model that incorporates contextual information from adjacent sections of the brain. This model integrates three tissue patches from neighboring sections using a siamese network architecture with cross-attention mechanisms. Leveraging spatially aligned information across consecutive sections, our approach achieves a more anatomically coherent reconstruction.We demonstrate that our model significantly improves realism and anatomical plausibility of reconstructed cellular distributions, as measured by both cell density prediction and brain area classification tasks. The error in predicted cell density was reduced to below 5% across large inpainting regions, marking a notable improvement over previous approaches. The model reliably preserves tissue borders and reconstructs larger structures like blood vessels, which are crucial for accurate cytoarchitectonic mapping.These findings underscore the potential of generative deep learning models for cytoarchitectonic research, opening new avenues for the automated reconstruction of histological data. Beyond inpainting small regions, our approach paves the way for the reconstruction of entirely missing brain slices, offering a powerful tool for bridging data gaps in high-resolution brain mapping efforts.


Contributing Institute(s):
  1. Strukturelle und funktionelle Organisation des Gehirns (INM-1)
Research Program(s):
  1. 5254 - Neuroscientific Data Analytics and AI (POF4-525) (POF4-525)
  2. Helmholtz AI - Helmholtz Artificial Intelligence Coordination Unit – Local Unit FZJ (E.40401.62) (E.40401.62)
  3. HIBALL - Helmholtz International BigBrain Analytics and Learning Laboratory (HIBALL) (InterLabs-0015) (InterLabs-0015)
  4. EBRAINS 2.0 - EBRAINS 2.0: A Research Infrastructure to Advance Neuroscience and Brain Health (101147319) (101147319)
  5. JL SMHB - Joint Lab Supercomputing and Modeling for the Human Brain (JL SMHB-2021-2027) (JL SMHB-2021-2027)

Appears in the scientific report 2025
Click to display QR Code for this record

The record appears in these collections:
Document types > Presentations > Poster
Institute Collections > INM > INM-1
Workflow collections > Public records
Publications database

 Record created 2025-06-27, last modified 2025-07-16



Rate this document:

Rate this document:
1
2
3
 
(Not yet reviewed)