(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
Abstract
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.
Data-driven polarimetric imaging: a review
Kui Yang, Fei Liu, Shiyang Liang, Meng Xiang, Pingli Han, Jinpeng Liu, Xue Dong, Yi Wei, Bingjian Wang, Koichi Shimizu, Xiaopeng Shao
Opto-Electronic Science
2024-02-24
Inverse design for material anisotropy and its application for a compact X-cut TFLN on-chip wavelength demultiplexer
Jiangbo Lyu, Tao Zhu, Yan Zhou, Zhenmin Chen, Yazhi Pi, Zhengtong Liu, Xiaochuan Xu, Ke Xu, Xu Ma, Lei Wang, Zizheng Cao, Shaohua Yu
Opto-Electronic Science
2024-01-09
Flat soliton microcomb source
Xinyu Wang, Xuke Qiu, Mulong Liu, Feng Liu, Mengmeng Li, Linpei Xue, Bohan Chen, Mingran Zhang, Peng Xie
Opto-Electronic Science
2023-12-29