Residential College | false |
Status | 已發表Published |
Cross-City Multi-Granular Adaptive Transfer Learning for Traffic Flow Prediction | |
Mo, Jiqian; Gong, Zhiguo | |
2023-11-01 | |
Source Publication | IEEE Transactions on Knowledge and Data Engineering |
ISSN | 1041-4347 |
Volume | 35Issue:11Pages:11246-11258 |
Abstract | Accurate traffic prediction is one of the most important techniques in building a smart city. Many works, especially deep learning models, have made great progress in traffic prediction based on rich historical data. However, many cities still suffer from the problem of data scarcity in many aspects. Some works use transfer learning to solve this kind of problem, but what and how to transfer is still an important problem. In this article, we propose a novel Cross-city Multi-Granular Adaptive Transfer Learning method named MGAT for traffic prediction with only a few data in the target city. We first use the meta-learning algorithm to train the model on multiple source cities to get a good initialization. And at the same time, the multi-granular regional characteristics of each source city will be obtained based on our model structure. Then we design an Adaptive Transfer module mainly composed of Spatial-Attention and Multi-head Attention mechanism to automatically select the most appropriate features from the multi-granular features trained from multiple source cities, to achieve the best transfer effect. We conduct extensive experiments on two kinds of real-world traffic datasets cross several cities. Experimental results with other state-of-the-art models demonstrate the effectiveness of the proposed model. |
Keyword | Attention Meta-learning Traffic Flow Prediction Transfer Learning |
DOI | 10.1109/TKDE.2022.3232185 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Engineering |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Information Systems ; Engineering, Electrical & Electronic |
WOS ID | WOS:001089176900024 |
Publisher | IEEE Computer Society |
Scopus ID | 2-s2.0-85146217328 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Gong, Zhiguo |
Affiliation | University of Macau, State Key Laboratory of Internet of Things for Smart City, Department of Computer and Information Science, Guangdong-Macau Joint Laboratory for Advanced and Intelligent Computing, Macao |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Mo, Jiqian,Gong, Zhiguo. Cross-City Multi-Granular Adaptive Transfer Learning for Traffic Flow Prediction[J]. IEEE Transactions on Knowledge and Data Engineering, 2023, 35(11), 11246-11258. |
APA | Mo, Jiqian., & Gong, Zhiguo (2023). Cross-City Multi-Granular Adaptive Transfer Learning for Traffic Flow Prediction. IEEE Transactions on Knowledge and Data Engineering, 35(11), 11246-11258. |
MLA | Mo, Jiqian,et al."Cross-City Multi-Granular Adaptive Transfer Learning for Traffic Flow Prediction".IEEE Transactions on Knowledge and Data Engineering 35.11(2023):11246-11258. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment