INSTANCE-BASED DEEP TRANSFER LEARNING WITH ATTENTION FOR STOCK MOVEMENT PREDICTION

Abstract

STOCK MOVEMENT PREDICTION IS ONE OF THE MOST CHALLENGING PROBLEMS IN TIME SERIES ANALYSIS DUE TO THE STOCHASTIC NATURE OF FINANCIAL MARKETS. IN RECENT YEARS, A PLETHORA OF STATISTICAL METHODS AND MACHINE LEARNING ALGORITHMS WERE PROPOSED FOR STOCK MOVEMENT PREDICTION. SPECIFICALLY, DEEP LEARNING MODELS ARE INCREASINGLY APPLIED FOR THE PREDICTION OF STOCK MOVEMENT. THE SUCCESS OF DEEP LEARNING MODELS RELIES ON THE ASSUMPTION THAT MASSIVE TRAINING DATA ARE AVAILABLE. HOWEVER, THIS ASSUMPTION IS IMPRACTICAL FOR STOCK MOVEMENT PREDICTION. IN STOCK MARKETS, A LARGE NUMBER OF STOCKS DO NOT HAVE ENOUGH HISTORICAL DATA, ESPECIALLY FOR THE COMPANIES WHICH UNDERWENT INITIAL PUBLIC OFFERING IN RECENT YEARS. IN THESE SITUATIONS, THE ACCURACY OF DEEP LEARNING MODELS TO PREDICT THE STOCK MOVEMENT COULD BE AFFECTED. TO ADDRESS THIS PROBLEM, IN THIS PAPER, WE PROPOSE NOVEL INSTANCE-BASED DEEP TRANSFER LEARNING MODELS WITH ATTENTION MECHANISM. IN THE EXPERIMENTS, WE COMPARE OUR PROPOSED METHODS WITH STATE-OF-THE-ART PREDICTION MODELS. EXPERIMENTAL RESULTS ON THREE PUBLIC DATASETS REVEAL THAT OUR PROPOSED METHODS SIGNIFICANTLY IMPROVE THE PERFORMANCE OF DEEP LEARNING MODELS WHEN LIMITED TRAINING DATA ARE AVAILABLE.