ehosseiniasl / 3d-convolutional-network

Alzheimer disease diagnosis by Deeply Supervised 3D Convolutional Network
80 stars 34 forks source link

Training of CAE #1

Open shubhamjain0594 opened 7 years ago

shubhamjain0594 commented 7 years ago

I am trying to implement your paper in Torch and wanted some help in it.

Firstly I want to know how are you exactly training the CAE, greedy layerwise or end to end.

ehosseiniasl commented 7 years ago

CAE are trained greedy layer-wise. The encoding layers are stacked and frozen during fine-tuning

shubhamjain0594 commented 7 years ago

Thanks for the reply.

Also you mention about pre-training, initial training and fine-tuning. I understand the part about pre-training on CADDementia and fine-tuning on ADNI but what do you mean by initial training of lower convolutional layers and how do you do it.

ehosseiniasl commented 7 years ago

there is no initial training! only pertraining and then fine-tuning

shubhamjain0594 commented 7 years ago

One last question, you mention about using pooling and conv in a CAE, so is your architecture Input -> Full Conv -> MaxPool -> MaxUnpool -> Validconv -> Output

Because you mention for input size of 200*150*150 you get feature maps of size 102*76*76, 52*40*40 and 28*22*22.

shubhamjain0594 commented 7 years ago

Also, can you add the list of ADNI files you have used? ADNI Data has 3000+ MRIs so it will be great if you can share the list of files you have used.

ehosseiniasl commented 7 years ago

@shubhamjain0594 In the current implementation of CAE, no pooling layer is used. to get hidden activation of a CAE to train the subsequent CAE, max_pooling is used. you can check the do_pretraining_cae() in main.py. Each cae has full_conv --> valid_conv structure.

ehosseiniasl commented 7 years ago

@shubhamjain0594 list of ADNI subject IDs are added!

christiantinauer commented 6 years ago

@ehosseiniasl Could you please also add the image IDs? Because the infos are ambiguous for some of the patients.

mjachowdhury commented 3 years ago

@ehosseiniasl Hi, I Would like to know that the subject ID, e.g., 002_S_0619, has 36 images for AD on different weighting, slice thickness, and acquisition. Did you select all the images for that subject ID or only one image from that subject ID as input for the model? I look forward to hearing from you. Thank you