计算机科学 ›› 2017, Vol. 44 ›› Issue (Z11): 106-109.doi: 10.11896/j.issn.1002-137X.2017.11A.021

• 智能计算 • 上一篇    下一篇

基于多输出学习的沪深300指数预测研究

唐艳琴,潘志松,张艳艳   

  1. 解放军理工大学指挥信息系统学院 南京210007,解放军理工大学指挥信息系统学院 南京210007,解放军理工大学指挥信息系统学院 南京210007
  • 出版日期:2018-12-01 发布日期:2018-12-01
  • 基金资助:
    本文受国家自然科学基金项目(61473149)资助

CSI 300 Index Prediction Research Based on Multi-output Learning

TANG Yan-qin, PAN Zhi-song and ZHANG Yan-yan   

  • Online:2018-12-01 Published:2018-12-01

摘要: 在股票市场中,人们通常会依赖于股票的历史交易数据来进行推测。目前采用SVM方法进行预测的研究较多,但其模型复杂,耗时较长,而且通常只预测未来1天的数据。文中采用多输出的正则化方法来预测未来多天的走势,对多任务的学习方法进行改进,提出了一种基于多输出的学习方法。实验表明,与SVM支持向量机的方法相比,该方法在沪深300指数预测的均方差值上提高了约10倍,运行时长也减少了近3/4。

关键词: 多输出学习,回归,股票预测,任务相关性

Abstract: In the stock market,people usually depend on the historical trading data to predict the trend of future.The method of SVM is more common in the stock prediction,but it is complex and time-consuming,and usually predicts the trend of one day.This article adopted regularization method of multi-output learning to predict the trend of many days.We improved the method of multi-task learning and put forward the method based on multi-output learning.Experiments in CSI 300 index show that the mean square error(MSE) of prediction in this method is about 10 times compared to that in the support vector machine (SVM) method,and the time consuming is also reduced nearly three-quarters.

Key words: Multi-output learning,Regression,Stock prediction,Task correlation

[1] BIELZA C,LI G,LARRANAGA P.Multi-dimensional classification with Bayesian networks[J].International Journal of Approximate Reasoning,2011,52:705-727.
[2] TIBSHIRANI R.Regression shrinkage and selection via the lasso[J].Journal of the Royal Statistical Society,2011,73(3):267-288.
[3] ZHOU J,CHEN J,YE J.MALSAR:Multi-tAsk Learning viaStructurAl Regularization[D].Arizona State University,2012.
[4] 肖瑞.不确定性时间序列的降维与相似性匹配研究[D].上海:东华大学,2014.
[5] 王琼瑶.基于改进的支持向量机技术在股票短期价格预测中的应用[D].重庆:重庆交通大学,2015.
[6] NG A.Stanford Engineering Everywhere CS229 lecture-MachineLearning.https://see.stanford.edu/course/cs229.
[7] MURPHY K P.Machine Learning:A Probabilistic Perspective[M].MIT Press,2012.
[8] BECK A,TEBOULLE M.A fast iterative shrinkage-thresholding algorithm for linear inverse problems[J].Siam Journal on Imaging Sciences,2009,2(1):183-202.
[9] HAN L,ZHANG Y.Multi-stage multi-task learning with re-duced rank[C]∥Proceedings of the Thirtieth AAAI Conference on Artificial Intelligence(AAAI-16).2016.
[10] LEE G,YANG E,HWANG S J.Asymmetric multi-task learning based on task relatedness and loss[C]∥Proceedings of the 33rd International Conference on Machine Learning.New York,NY,USA,2016 :230-238.
[11] 吴喜之,刘苗编.应用时间序列分析:R软件陪同[M].北京:机械工业出版社,2014.
[12] TASKAR B.Learning structured prediction models:A large marginapproach[D].Stanford University,2004.
[13] JOACHIMS T,HOFMANN T,YUE Y S,et al.Predicting Struc-tured Objects with Support Vector Machines[J].Communications of the ACM,2009,52(11):97-104.
[14] JOACHIMS T,FINLEY T,YU C N J.Cutting-plane training of structural svms[J].Machine Learning,2009,77(1):27-59.
[15] HAGAN M T,DEMUTH HB,BEALE M H,et al.Neural network design[M].PWS Publishing Company Boston,1996.
[16] MAGERMAN D M.Statistical decision-tree models for parsing[C]∥Proceedings of the 33rd Annual Meeting on Association for Computational Linguistics.1995:276-283.
[17] DINUZZO F,ONG C S,GEHLER P,et al.Learning output kernels with block coordinate descent[C]∥Proceedings of the 28th Annual International Conference on Machine Learning.Bellevue,WA,USA,2011.
[18] DINUZZO F,FUKUMIZU K.Learning low-rank output kernels[J].Journal of Machine Learning Research Proceedings Track ,2011(20):181-196.
[19] ZHANG M L,ZHOU Z H.A Review on Multi-Label Learning Algorithms[J].IEEE Transactions on Knowledge & Data Engineering,2014,6(8):1819-1837.

No related articles found!
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!