(Preprint) Retrieval & Interaction Machine for Tabular Data Prediction
Jiarui Qin 秦佳锐 ¹, Weinan Zhang 张伟楠 ¹, Rong Su ², Zhirong Liu ², Weiwen Liu ², Ruiming Tang 唐睿明 ², Xiuqiang He 何秀强 ², Yong Yu 俞勇 ¹
¹ Shanghai Jiao Tong University
² Huawei Noah’s Ark Lab
arXiv, 2021-08-11

Prediction over tabular data is an essential task in many data science applications such as recommender systems, online advertising, medical treatment, etc. Tabular data is structured into rows and columns, with each row as a data sample and each column as a feature attribute. Both the columns and rows of the tabular data carry useful patterns that could improve the model prediction performance. However, most existing models focus on the cross-column patterns yet overlook the cross-row patterns as they deal with single samples independently.

In this work, we propose a general learning framework named Retrieval & Interaction Machine (RIM) that fully exploits both cross-row and cross-column patterns among tabular data. Specifically, RIM first leverages search engine techniques to efficiently retrieve useful rows of the table to assist the label prediction of the target row, then uses feature interaction networks to capture the cross-column patterns among the target row and the retrieved rows so as to make the final label prediction.

We conduct extensive experiments on 11 datasets of three important tasks, i.e., CTR prediction (classification), top-n recommendation (ranking) and rating prediction (regression). Experimental results show that RIM achieves significant improvements over the state-of-the-art and various baselines, demonstrating the superiority and efficacy of RIM.
Retrieval & Interaction Machine for Tabular Data Prediction_1
Retrieval & Interaction Machine for Tabular Data Prediction_2
Retrieval & Interaction Machine for Tabular Data Prediction_3
Retrieval & Interaction Machine for Tabular Data Prediction_4
  • Power grid fault diagnosis based on a deep pyramid convolutional neural network
  • Xu Zhang 张旭, Huiting Zhang, Dongying Zhang, Yixian Wang, Ruiting Ding, Yuchuan Zheng, Yongxu Zhang
  • CSEE Journal of Power and Energy Systems
  • 2022-05-06
  • China's factor reallocation effect considering energy
  • Guangqing Xu, Xiaoyu Chen
  • Chinese Journal of Population, Resources and Environment
  • 2022-05-02
  • Cannabidiol prevents depressive-like behaviors through the modulation of neural stem cell differentiation
  • Ming Hou, Suji Wang, Dandan Yu, Xinyi Lu, Xiansen Zhao, Zhangpeng Chen, Chao Yan
  • Frontiers of Medicine
  • 2022-04-26
  • Cultivation of gut microorganisms of the marine ascidian Halocynthia roretzi reveals their potential roles in the environmental adaptation of their host
  • Yang Yang, Yuting Zhu, Haiming Liu, Jiankai Wei, Haiyan Yu, Bo Dong
  • Marine Life Science & Technology
  • 2022-04-26
  • Data network traffic analysis and optimization strategy of real-time power grid dynamic monitoring system for wide-frequency measurements
  • Jinsong Li, Hao Liu, Wenzhuo Li, Tianshu Bi, Mingyang Zhao
  • Global Energy Interconnection
  • 2022-04-25
  • Field distribution of the Z₂ topological edge state revealed by cathodoluminescence nanoscopy
  • Xiao He, Donglin Liu, Hongfei Wang, Liheng Zheng, Bo Xu, Biye Xie, Meiling Jiang, Zhixin Liu, Jin Zhang, Minghui Lu, Zheyu Fang
  • Opto-Electronic Advances
  • 2022-04-25
  • Advances in femtosecond laser direct writing of fiber Bragg gratings in multicore fibers: technology, sensor and laser applications
  • Alexey Wolf, Alexander Dostovalov, Kirill Bronnikov, Mikhail Skvortsov, Stefan Wabnitz, Sergey Babin
  • Opto-Electronic Advances
  • 2022-04-25
  • Graphene-empowered dynamic metasurfaces and metadevices
  • Chao Zeng, Hua Lu, Dong Mao, Yueqing Du, He Hua, Wei Zhao, Jianlin Zhao
  • Opto-Electronic Advances
  • 2022-04-25
  • Charge carrier dynamics in different crystal phases of CH₃NH₃PbI₃ perovskite
  • Efthymis Serpetzoglou, Ioannis Konidakis, George Kourmoulakis, Ioanna Demeridou, Konstantinos Chatzimanolis, Christos Zervos, George Kioseoglou, Emmanuel Kymakis, Emmanuel Stratakis
  • Opto-Electronic Science
  • 2022-04-21
  • Applications of optically and electrically driven nanoscale bowtie antennas
  • Zhongjun Jiang, Yingjian Liu, Liang Wang
  • Opto-Electronic Science
  • 2022-04-20
  • Validation of the bodily expressive action stimulus test among Chinese adults and children
  • Yunmei Yang, Wenwen Hou, Jing Li
  • PsyCh Journal
  • 2022-04-17

  • A New Interpolation Approach and Corresponding Instance-Based Learning                                Boosting the Generalization Capability in Cross-Domain Few-shot Learning via Noise-enhanced Supervised Autoencoder
    Copyright © PubCard