arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.02676cs.LGstat.ML
收藏

Exchangeable Neural ODE for Set Modeling

UNC-Chapel Hill(北卡罗来纳大学教堂山分校);The University of North Carolina(北卡罗来纳大学)

Yang Li, Haidong Yi, Christopher M. Bender, Siyuan Shan, Junier B. Oliva

英文摘要:

Reasoning over an instance composed of a set of vectors, like a point cloud, requires that one accounts for intra-set dependent features among elements. However, since such instances are unordered, the elements' features should remain unchanged when the input's order is permuted. This property, permutation equivariance, is a challenging constraint for most neural architectures. While recent work has proposed global pooling and attention-based solutions, these may be limited in the way that intradependencies are captured in practice. In this work we propose a more general formulation to achieve permutation equivariance through ordinary differential equations (ODE). Our proposed module, Exchangeable Neural ODE (ExNODE), can be seamlessly applied for both discriminative and generative tasks. We also extend set modeling in the temporal dimension and propose a VAE based model for temporal set modeling. Extensive experiments demonstrate the efficacy of our method over strong baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑