Abstract:
Open-domain dialog generation, which is a crucial component of artificial intelligence, is an essential and challenging problem. In this article, we present a personalize...Show MoreMetadata
Abstract:
Open-domain dialog generation, which is a crucial component of artificial intelligence, is an essential and challenging problem. In this article, we present a personalized dialog system, which leverages the advantages of multitask learning and reinforcement learning for personalized dialogue generation (MRPDG). Specifically, MRPDG consists of two subtasks: 1) an author profiling module that recognizes user characteristics from the input sentence (auxiliary task) and 2) a personalized dialog generation system that generates informative, grammatical, and coherent responses with reinforcement learning algorithms (primary task). Three kinds of rewards are proposed to generate high-quality conversations. We investigate the effectiveness of three widely used reinforcement learning methods [i.e., Q-learning, policy gradient, and actor-critic (AC) algorithm] in a personalized dialog generation system and demonstrate that the AC algorithm achieves the best results on the underlying framework. Comprehensive experiments are conducted to evaluate the performance of the proposed model on two real-life data sets. Experimental results illustrate that MRPDG is able to produce high-quality personalized dialogs for users with different characteristics. Quantitatively, the proposed model can achieve better performance than the compared methods across different evaluation metrics, such as the human evaluation, BiLingual Evaluation Understudy (BLEU), and perplexity.
Published in: IEEE Transactions on Neural Networks and Learning Systems ( Volume: 32, Issue: 1, January 2021)
Funding Agency:

Chinese Academy of Sciences, Shenzhen Institutes of Advanced Technology, Shenzhen, China
Min Yang received the Ph.D. degree from The University of Hong Kong, Hong Kong, in 2017.
She is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. Her current research interests include machine learning, deep learning, and natural language processing.
Min Yang received the Ph.D. degree from The University of Hong Kong, Hong Kong, in 2017.
She is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. Her current research interests include machine learning, deep learning, and natural language processing.View more

Chinese Academy of Sciences, Shenzhen Institutes of Advanced Technology, Shenzhen, China
Weiyi Huang received the B.S. degree from the University of Science and Technology of China, Hefei, China, in 2017. She is currently pursuing the M.Sc. degree with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China.
Her research interests include natural language processing, information retrieval, and data mining.
Weiyi Huang received the B.S. degree from the University of Science and Technology of China, Hefei, China, in 2017. She is currently pursuing the M.Sc. degree with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China.
Her research interests include natural language processing, information retrieval, and data mining.View more

Department of Computer Science, Shanghai University of Finance and Economics, Shanghai, China
Wenting Tu received the M.S. degree from the Department of Computer Science and Technology, East China Normal University, Shanghai, China, in 2012, and the Ph.D. degree in computer science from The University of Hong Kong, Hong Kong, in 2016.
She is currently an Associate Professor with the Department of Computer Science, Shanghai University of Finance and Economics, Shanghai. Her research interests include data mining, ma...Show More
Wenting Tu received the M.S. degree from the Department of Computer Science and Technology, East China Normal University, Shanghai, China, in 2012, and the Ph.D. degree in computer science from The University of Hong Kong, Hong Kong, in 2016.
She is currently an Associate Professor with the Department of Computer Science, Shanghai University of Finance and Economics, Shanghai. Her research interests include data mining, ma...View more

Chinese Academy of Sciences, Shenzhen Institutes of Advanced Technology, Shenzhen, China
Qiang Qu received the M.Sc. degree in computer science from Peking University, Beijing, China, in 2011, and the Ph.D. degree from Aarhus University, Aarhus, Denmark, in 2014.
He is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. His Ph.D. study was supported by the GEOCrowd Project under Marie Skodowska-Curie Actions. His current research i...Show More
Qiang Qu received the M.Sc. degree in computer science from Peking University, Beijing, China, in 2011, and the Ph.D. degree from Aarhus University, Aarhus, Denmark, in 2014.
He is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. His Ph.D. study was supported by the GEOCrowd Project under Marie Skodowska-Curie Actions. His current research i...View more

School of Intelligent Systems Engineering, Sun Yat-sen University, Guangzhou, China
Ying Shen received the Erasmus Mundus degree (master’s) in natural language processing from the University of Franche-Comté, Besançon, France, and the University of Wolverhampton, Wolverhampton, U.K., and the Ph.D. degree in medical and biomedical information science from the University of Paris Ouest Nanterre La Défense, Nanterre, France.
She is currently an Associate Professor with the School of Intelligent Systems Engin...Show More
Ying Shen received the Erasmus Mundus degree (master’s) in natural language processing from the University of Franche-Comté, Besançon, France, and the University of Wolverhampton, Wolverhampton, U.K., and the Ph.D. degree in medical and biomedical information science from the University of Paris Ouest Nanterre La Défense, Nanterre, France.
She is currently an Associate Professor with the School of Intelligent Systems Engin...View more

Shenzhen Key Laboratory for Information Centric Networking and Blockchain Technology, Peking University, Shenzhen, China
Peng Cheng Laboratory, PCL Research Center of Networks and Communications, Shenzhen, China
Kai Lei received the B.Sc. degree in computer science from Peking University, Beijing, China, in 1998, the M.Sc. degree in computer science from Columbia University, New York, NY, USA, in 1999, and the Ph.D. degree in computer science from Peking University in 2015.
He has worked at companies, including IBM Thomas J. Watson Research Center, Ossining, NY, USA, Citigroup, New York, NY, USA, Oracle, Redwood City, CA, USA, and...Show More
Kai Lei received the B.Sc. degree in computer science from Peking University, Beijing, China, in 1998, the M.Sc. degree in computer science from Columbia University, New York, NY, USA, in 1999, and the Ph.D. degree in computer science from Peking University in 2015.
He has worked at companies, including IBM Thomas J. Watson Research Center, Ossining, NY, USA, Citigroup, New York, NY, USA, Oracle, Redwood City, CA, USA, and...View more

Chinese Academy of Sciences, Shenzhen Institutes of Advanced Technology, Shenzhen, China
Min Yang received the Ph.D. degree from The University of Hong Kong, Hong Kong, in 2017.
She is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. Her current research interests include machine learning, deep learning, and natural language processing.
Min Yang received the Ph.D. degree from The University of Hong Kong, Hong Kong, in 2017.
She is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. Her current research interests include machine learning, deep learning, and natural language processing.View more

Chinese Academy of Sciences, Shenzhen Institutes of Advanced Technology, Shenzhen, China
Weiyi Huang received the B.S. degree from the University of Science and Technology of China, Hefei, China, in 2017. She is currently pursuing the M.Sc. degree with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China.
Her research interests include natural language processing, information retrieval, and data mining.
Weiyi Huang received the B.S. degree from the University of Science and Technology of China, Hefei, China, in 2017. She is currently pursuing the M.Sc. degree with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China.
Her research interests include natural language processing, information retrieval, and data mining.View more

Department of Computer Science, Shanghai University of Finance and Economics, Shanghai, China
Wenting Tu received the M.S. degree from the Department of Computer Science and Technology, East China Normal University, Shanghai, China, in 2012, and the Ph.D. degree in computer science from The University of Hong Kong, Hong Kong, in 2016.
She is currently an Associate Professor with the Department of Computer Science, Shanghai University of Finance and Economics, Shanghai. Her research interests include data mining, machine learning, and natural language processing.
Wenting Tu received the M.S. degree from the Department of Computer Science and Technology, East China Normal University, Shanghai, China, in 2012, and the Ph.D. degree in computer science from The University of Hong Kong, Hong Kong, in 2016.
She is currently an Associate Professor with the Department of Computer Science, Shanghai University of Finance and Economics, Shanghai. Her research interests include data mining, machine learning, and natural language processing.View more

Chinese Academy of Sciences, Shenzhen Institutes of Advanced Technology, Shenzhen, China
Qiang Qu received the M.Sc. degree in computer science from Peking University, Beijing, China, in 2011, and the Ph.D. degree from Aarhus University, Aarhus, Denmark, in 2014.
He is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. His Ph.D. study was supported by the GEOCrowd Project under Marie Skodowska-Curie Actions. His current research interest includes large-scale data management and mining.
Qiang Qu received the M.Sc. degree in computer science from Peking University, Beijing, China, in 2011, and the Ph.D. degree from Aarhus University, Aarhus, Denmark, in 2014.
He is currently an Associate Professor with the Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen, China. His Ph.D. study was supported by the GEOCrowd Project under Marie Skodowska-Curie Actions. His current research interest includes large-scale data management and mining.View more

School of Intelligent Systems Engineering, Sun Yat-sen University, Guangzhou, China
Ying Shen received the Erasmus Mundus degree (master’s) in natural language processing from the University of Franche-Comté, Besançon, France, and the University of Wolverhampton, Wolverhampton, U.K., and the Ph.D. degree in medical and biomedical information science from the University of Paris Ouest Nanterre La Défense, Nanterre, France.
She is currently an Associate Professor with the School of Intelligent Systems Engineering, Sun Yat-sen University, Guangzhou, China. Her research interests include medical informatics, natural language processing, and machine learning.
Ying Shen received the Erasmus Mundus degree (master’s) in natural language processing from the University of Franche-Comté, Besançon, France, and the University of Wolverhampton, Wolverhampton, U.K., and the Ph.D. degree in medical and biomedical information science from the University of Paris Ouest Nanterre La Défense, Nanterre, France.
She is currently an Associate Professor with the School of Intelligent Systems Engineering, Sun Yat-sen University, Guangzhou, China. Her research interests include medical informatics, natural language processing, and machine learning.View more

Shenzhen Key Laboratory for Information Centric Networking and Blockchain Technology, Peking University, Shenzhen, China
Peng Cheng Laboratory, PCL Research Center of Networks and Communications, Shenzhen, China
Kai Lei received the B.Sc. degree in computer science from Peking University, Beijing, China, in 1998, the M.Sc. degree in computer science from Columbia University, New York, NY, USA, in 1999, and the Ph.D. degree in computer science from Peking University in 2015.
He has worked at companies, including IBM Thomas J. Watson Research Center, Ossining, NY, USA, Citigroup, New York, NY, USA, Oracle, Redwood City, CA, USA, and Google, Mountain View, CA, USA, from 1999 to 2004. He is currently an Associate Professor with the School of Electronics and Computer Engineering, Peking University. His research interests include named data networking, blockchain, and big data technologies.
Kai Lei received the B.Sc. degree in computer science from Peking University, Beijing, China, in 1998, the M.Sc. degree in computer science from Columbia University, New York, NY, USA, in 1999, and the Ph.D. degree in computer science from Peking University in 2015.
He has worked at companies, including IBM Thomas J. Watson Research Center, Ossining, NY, USA, Citigroup, New York, NY, USA, Oracle, Redwood City, CA, USA, and Google, Mountain View, CA, USA, from 1999 to 2004. He is currently an Associate Professor with the School of Electronics and Computer Engineering, Peking University. His research interests include named data networking, blockchain, and big data technologies.View more