CASED-Tensorflow

Tensorflow implementation of Curriculum Adaptive Sampling for Extreme Data Imbalance with multi GPU using LUNA16

Preprocessing Tutorial

• convert_luna_to_npy
• create patch

Usage for preprocessing

> python all_in_one.py

• Check src_root and save_path

Usage for train

> python main_train.py

• See main_train.py for other arguments.

Usage for test

> python main_test.py

Issue

• The hyper-parameter information is not listed in the paper, so I'm still testing it.
• Use Snapshot Ensemble (M=10, init_lr=0.1)
• Or Fix learning rate 0.01

def Snapshot(t, T, M, alpha_zero) :
    """
    t = # of current iteration
    T = # of total iteration
    M = # of snapshot
    alpha_zero = init learning rate
    """

    x = (np.pi * (t % (T // M))) / (T // M)
    x = np.cos(x) + 1

    lr = (alpha_zero / 2) * x

    return lr

Summary

Preprocessing

• Resample

> 1.25mm

• Hounsfield

> minHU = -1000
> maxHU = 400

• Zero centering

> Pixel Mean = 0.25

Network Architecture

Algorithm

Result

Author

Junho Kim / @Lunit