Guoqiang Shen, Zhiqiang Zou, A-Li Luo, Shuxin Hong, Xiao Kong
{"title":"基于动量对比学习的星系形态分类模型","authors":"Guoqiang Shen, Zhiqiang Zou, A-Li Luo, Shuxin Hong, Xiao Kong","doi":"10.1088/1538-3873/acf8f7","DOIUrl":null,"url":null,"abstract":"Abstract The taxonomy of galaxy morphology plays an important role in astrophysics and provides great help for the study of galaxy evolution. To integrate the advantages of unsupervised learning without labels and supervised learning with high classification accuracy, this paper proposes a galaxy morphology classification model based on a momentum contrastive learning algorithm named Momentum Contrastive Learning Galaxy (MCL-Galaxy), which mainly includes two parts (i) pre-training of the model, where the ResNet_50 backbone network acts as an encoder to learn the galaxy morphology image features, which are stored in the queue and their consistency is ensured by using the momentum contrastive learning algorithm; and (ii) transfer learning, where Mahalanobis distance can assist in improving classification accuracy in downstream tasks where both encoder and queue are transferred. To evaluate the performance of MCL-Galaxy, we use the data set of the Galaxy Zoo challenge project on Kaggle for comparative testing. The experimental results show that the classification accuracy of MCL-Galaxy can reach 90.12%, which is 8.12% higher than the unsupervised state-of-the-art results. Although it is 3.1% lower than the advanced supervised method, it has the advantage of no label and can achieve a higher accuracy rate at the first epoch of classification iteration. This suggests that the gap between unsupervised and supervised representation learning in the field of Galaxy Morphologies classification tasks is well bridged.","PeriodicalId":20820,"journal":{"name":"Publications of the Astronomical Society of the Pacific","volume":"50 1","pages":"0"},"PeriodicalIF":3.3000,"publicationDate":"2023-10-01","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"A Galaxy Morphology Classification Model Based on Momentum Contrastive Learning\",\"authors\":\"Guoqiang Shen, Zhiqiang Zou, A-Li Luo, Shuxin Hong, Xiao Kong\",\"doi\":\"10.1088/1538-3873/acf8f7\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"Abstract The taxonomy of galaxy morphology plays an important role in astrophysics and provides great help for the study of galaxy evolution. To integrate the advantages of unsupervised learning without labels and supervised learning with high classification accuracy, this paper proposes a galaxy morphology classification model based on a momentum contrastive learning algorithm named Momentum Contrastive Learning Galaxy (MCL-Galaxy), which mainly includes two parts (i) pre-training of the model, where the ResNet_50 backbone network acts as an encoder to learn the galaxy morphology image features, which are stored in the queue and their consistency is ensured by using the momentum contrastive learning algorithm; and (ii) transfer learning, where Mahalanobis distance can assist in improving classification accuracy in downstream tasks where both encoder and queue are transferred. To evaluate the performance of MCL-Galaxy, we use the data set of the Galaxy Zoo challenge project on Kaggle for comparative testing. The experimental results show that the classification accuracy of MCL-Galaxy can reach 90.12%, which is 8.12% higher than the unsupervised state-of-the-art results. Although it is 3.1% lower than the advanced supervised method, it has the advantage of no label and can achieve a higher accuracy rate at the first epoch of classification iteration. This suggests that the gap between unsupervised and supervised representation learning in the field of Galaxy Morphologies classification tasks is well bridged.\",\"PeriodicalId\":20820,\"journal\":{\"name\":\"Publications of the Astronomical Society of the Pacific\",\"volume\":\"50 1\",\"pages\":\"0\"},\"PeriodicalIF\":3.3000,\"publicationDate\":\"2023-10-01\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Publications of the Astronomical Society of the Pacific\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1088/1538-3873/acf8f7\",\"RegionNum\":3,\"RegionCategory\":\"物理与天体物理\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q2\",\"JCRName\":\"ASTRONOMY & ASTROPHYSICS\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Publications of the Astronomical Society of the Pacific","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1088/1538-3873/acf8f7","RegionNum":3,"RegionCategory":"物理与天体物理","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q2","JCRName":"ASTRONOMY & ASTROPHYSICS","Score":null,"Total":0}
A Galaxy Morphology Classification Model Based on Momentum Contrastive Learning
Abstract The taxonomy of galaxy morphology plays an important role in astrophysics and provides great help for the study of galaxy evolution. To integrate the advantages of unsupervised learning without labels and supervised learning with high classification accuracy, this paper proposes a galaxy morphology classification model based on a momentum contrastive learning algorithm named Momentum Contrastive Learning Galaxy (MCL-Galaxy), which mainly includes two parts (i) pre-training of the model, where the ResNet_50 backbone network acts as an encoder to learn the galaxy morphology image features, which are stored in the queue and their consistency is ensured by using the momentum contrastive learning algorithm; and (ii) transfer learning, where Mahalanobis distance can assist in improving classification accuracy in downstream tasks where both encoder and queue are transferred. To evaluate the performance of MCL-Galaxy, we use the data set of the Galaxy Zoo challenge project on Kaggle for comparative testing. The experimental results show that the classification accuracy of MCL-Galaxy can reach 90.12%, which is 8.12% higher than the unsupervised state-of-the-art results. Although it is 3.1% lower than the advanced supervised method, it has the advantage of no label and can achieve a higher accuracy rate at the first epoch of classification iteration. This suggests that the gap between unsupervised and supervised representation learning in the field of Galaxy Morphologies classification tasks is well bridged.
期刊介绍:
The Publications of the Astronomical Society of the Pacific (PASP), the technical journal of the Astronomical Society of the Pacific (ASP), has been published regularly since 1889, and is an integral part of the ASP''s mission to advance the science of astronomy and disseminate astronomical information. The journal provides an outlet for astronomical results of a scientific nature and serves to keep readers in touch with current astronomical research. It contains refereed research and instrumentation articles, invited and contributed reviews, tutorials, and dissertation summaries.