Deep Learning-Based Denoising of Monte Carlo Simulations for Efficient Laser-Tissue Interaction Modeling

Overview

This repository contains the official implementation of the paper "Deep Learning-Based Denoising of Monte Carlo Simulations for Efficient Laser-Tissue Interaction Modeling".

Features

• Denoising networks for 3D Monte Carlo simulation data: U-Net, Restormer, NAFNet
• Python codes for training, evaluation, loading dataset
• Pre-trained model weights

0. Installation

Clone this repository: git clone https://github.com/CMME-Lab/LIFUSimul-DL.git
Install all prerequisites with pip install -r requirements.txt

1. Preparing dataset

• Dataset will provided upon reasonable request.
• Locate the '.h5' files in your desired root directory.
• Specify the dataset path using the '--data_path' argument when running train/test scripts.

Example usage :
python train.py --data_path ./my_root_path

Custom dataset format

• Prepare two HDF5 files following the same structure used by dataset.py.

  • frate_high.h5: high-resolution target data.

  • frate_low_{sampling}.h5: corresponding low-sampling input data. {sampling} must match the value passed to --n_sampling.

• Both HDF5 files must contain the same dataset keys. For each key, the dataloader reads the paired data as fh = frate_high[key].

• Each dataset entry should be a 3D volume. The paired high- and low-resolution volumes must have the same shape before cropping, and each key must correspond to the same sample in both HDF5 files.

• The current train/validation/test split assumes 6 cases with 1000 samples per case. If your custom dataset uses a different number of cases or samples per case, update the num_cases and length arguments in split_dataset() accordingly.

2. Training

Run the training process using train.py.

Example usage :
python train.py --run_name my_experiments --modality ct --model swin --num_epoch 100 --decay_epoch 100 --init_model --cuda

3. Evaluation

Run the evaluation using test.py.

Example usage :
python test.py --run_name my_experiments --modality ct --model swin --cuda --plot

Authors

Minjee Seo, Seonaeng Cho, Minju Seol, Haris Ghafoor and Kyungho Yoon, School of Mathematics and Computing (Computational Science and Engineering), Yonsei University, Seoul, Republic of Korea

Acknowledgement

The work was supported by the National Research Foundation of Korea (NRF) funded by the Korean government (MSIT) under Grants RS-2024-00335185.

License

MIT License

Contact

For any queries, please reach out to Minjee Seo.