DOI QR코드

DOI QR Code

A Label Inference Algorithm Considering Vertex Importance in Semi-Supervised Learning

준지도 학습에서 꼭지점 중요도를 고려한 레이블 추론

  • 오병화 (서강대학교 컴퓨터공학과) ;
  • 양지훈 (서강대학교 컴퓨터공학과) ;
  • 이현진 (한국전자통신연구원 방송통신미디어연구소)
  • Received : 2015.07.23
  • Accepted : 2015.10.02
  • Published : 2015.12.15

Abstract

Abstract Semi-supervised learning is an area in machine learning that employs both labeled and unlabeled data in order to train a model and has the potential to improve prediction performance compared to supervised learning. Graph-based semi-supervised learning has recently come into focus with two phases: graph construction, which converts the input data into a graph, and label inference, which predicts the appropriate labels for unlabeled data using the constructed graph. The inference is based on the smoothness assumption feature of semi-supervised learning. In this study, we propose an enhanced label inference algorithm by incorporating the importance of each vertex. In addition, we prove the convergence of the suggested algorithm and verify its excellence.

준지도 학습은 기계 학습의 한 분야로서, 레이블된 데이터와 레이블되지 않은 데이터 모두를 사용하여 모델을 학습함으로써 지도 학습에 비해 예측 정확도를 높일 수 있다. 최근 각광받고 있는 그래프 기반 준지도 학습은 입력 데이터를 그래프의 형태로 변환하는 그래프 구축 단계와 이를 사용하여 레이블되지 않은 데이터의 레이블을 예측하는 레이블 추론 단계로 나뉜다. 이 추론은 준지도 학습에서의 평활도 가정을 기본으로 한다. 본 연구에서는 추가로 각 꼭지점 중요도를 결합함으로써 개선된 레이블 추론 알고리즘을 제안한다. 이와 함께 알고리즘의 수렴성을 증명하고, 또한 실험을 통해 알고리즘의 우수성을 검증하였다.

Keywords

Acknowledgement

Grant : 효과적인 데이터 분석을 위한 준지도 학습 알고리즘 개발, 상황인지형 Tele-Screen 시스템 기술 개발

Supported by : 한국연구재단, 정보통신기술진흥센터

References

  1. A. Subramanya and P. P. Talukdar, Graph-Based Semi-Supervised Learning, p.111, Morgan & Claypool, 2014.
  2. D. Kim, J. G. Joung, K. A. Sohn, H. Shin, Y. R. Park, M. D. Ritchie and J. H. Kim, "Knowledge boosting: a graph-based integration approach with multi-omics data and genomic knowledge for cancer clinical outcome prediction," Journal of the American Medical Informatics Association, Vol. 22, No. 1, pp. 109-120, 2015. https://doi.org/10.1136/amiajnl-2013-002481
  3. K. Park and H. Shin, "Stock price prediction based on a complex interrelation network of economic factors," Engineering Applications of Artificial Intelligence, Vol. 26, No. 5, pp. 1550-1561, 2013. https://doi.org/10.1016/j.engappai.2013.01.009
  4. O. Chapelle, B. Schölkopf and A Zien. Semi-Supervised Learning, p.508, Cambridge: MIT Press, 2006.
  5. X. Zhu, Z. Ghahramani and J. Lafferty. "Semi-Supervised Learning Using Gaussian Fields and Harmonic Functions," Proc. of the 20th International Conference on Machine Learning, pp. 912-919, 2003.
  6. X. Zhu and A. B. Goldberg. Introduction to Semi-Supervised Learning, p.116, Morgan & Claypool, 2009.
  7. D. Zhou, O. Bousquet, T. N. Lal, J. Wetson and B. Scholkopf, "Learning with Local and Global Consistency," Advances in Neural Information Processing Systems, Vol. 16, pp. 321-328, 2004.
  8. B. Dalvi, W. W. Cohen and J. Callan, "Exploratory Learning," Machine Learning and Knowledge Discovery in Databases, Springer Berlin Heidelberg, pp. 128-143, 2013.
  9. J. Lin, "Divergence Measures Based on the Shannon Entropy," IEEE Transactions on Information Theory, Vol. 37, No. 1, pp. 145-151, 1991. https://doi.org/10.1109/18.61115
  10. R. E. Fan, P. H. Chen and C. J. Lin, "Working Set Selection Using Second Order Information for Training Support Vector Machines," Journal of Machine Learning Research, Vol. 6, pp. 1889-1918, 2005.
  11. Y. Fang, K. C. C. Chang and H. W. Lauw. "Graphbased Semi-supervised Learning: Realizing Pointwise Smoothness Probabilistically," Proc. of the 20th International Conference on Machine Learning, 2014.