Deep Learning the Deep Sky: Recovering low surface brightness objects with machine learning
- New Candidates
- We are planning to update the catalogue of low mass dwarf galaxies to create the most complete and comprehensive survey of the local universe.
- A dataset that will be useful not only for galaxy physics but maybe also cosmology.
- Transfer Learning
- The ImageNet dataset is a very large collection of human annotated photographs designed by academics for developing computer vision algorithms.
- It has approximately 1 million images and 1,000 object classes.
- The most popular CNN architectures have been trained on this dataset and their associated weights are available for download.
- Can a machine that has been pre-trained to categorize everyday object also catagorize astronomical images?
- Transfer Learning
- Let's take a popular model "VGG16".
- It has 16 CNN layers.
- Outputs 4X4X512 neurons.
- Millions of weights.
- We take all of the trained weights and use the output prediction as an input into a fully connected DNN (512) and finally output a Single neuron activation for 'dE'.
- The pretrained network its not doing better than just doing our own training!
- Let's fine-tune! Unfreeze the VGG weights and train for an additional 10 epochs.
- We gain more than five% in accuracy in detecting dE with a network that has seen everyday objects - why?
- Answer: Non-astronomers can easily learn to categorize galaxy images.
- Fine-Tuning