Classification of stellar spectra with SVM based on within-class scatter and between-class scatter
Support Vector Machine (SVM) is a popular data mining technique, and it has been widely applied in astronomical tasks, especially in stellar spectra classification. Since SVM doesn’t take the data distribution into consideration, and therefore, its classification efficiencies can’t be greatly improved. Meanwhile, SVM ignores the internal information of the training dataset, such as the within-class structure and between-class structure. In view of this, we propose a new classification algorithm-SVM based on Within-Class Scatter and Between-Class Scatter (WBS-SVM) in this paper. WBS-SVM tries to find an optimal hyperplane to separate two classes. The difference is that it incorporates minimum within-class scatter and maximum between-class scatter in Linear Discriminant Analysis (LDA) into SVM. These two scatters represent the distributions of the training dataset, and the optimization of WBS-SVM ensures the samples in the same class are as close as possible and the samples in different classes are as far as possible. Experiments on the K-, F-, G-type stellar spectra from Sloan Digital Sky Survey (SDSS), Data Release 8 show that our proposed WBS-SVM can greatly improve the classification accuracies.
KeywordsSupport Vector Machine (SVM) Linear Discriminant Analysis (LDA) Within-class scatter Between-class scatter
This work was supported by the National Nature Science Foundation of China (61202311, U1731128), the Natural Science Foundation of Shanxi (201601D011042), the Program for Outstanding Innovative Team of High Learning Institutions of Shanxi and the Outstanding Youth Funds of North University of China.
- Bu, Y.D., Pan, J.C., Jiang, B., et al.: Publ. Astron. Soc. Jpn. 65(4), 173 (2013) Google Scholar
- Harinder, P.S., Gulati, R.K., Gupta, R.: Mon. Not. R. Astron. Soc. 295(2), 312 (2018) Google Scholar
- Jiang, B., Li, Z.X., Qu, M.X., et al.: Spectrosc. Spectr. Anal. 36(7), 2275 (2016) Google Scholar
- Qin, D.M., Hu, Z.Y., Zhao, Y.H.: Spectrosc. Spectr. Anal. 24(4), 507 (2004) Google Scholar