Universal approximation of functions on sets

Journal article

Modelling functions of sets, or equivalently, permutation-invariant functions, is a longstanding challenge in machine learning. Deep Sets is a popular method which is known to be a universal approximator for continuous set functions. We provide a theoretical analysis of Deep Sets which shows that this universal approximation property is only guaranteed if the model's latent space is sufficiently high-dimensional. If the latent space is even one dimension lower than necessary, there exist piecewise-afine functions for which Deep Sets performs no better than a nafive constant baseline, as judged by worst-case error. Deep Sets may be viewed as the most efficient incarnation of the Janossy pooling paradigm. We identify this paradigm as encompassing most currently popular set-learning methods. Based on this connection, we discuss the implications of our results for set learning more broadly, and identify some open questions on the universality of Janossy pooling in general.

Authors: Wagstaff, E; Fuchs, FB; Engelcke, M; Osborne, MA; Posner, I

• Publication status: Published
• Peer review status: Peer reviewed
• Publisher: Journal of Machine Learning Research
• Journal: Journal of Machine Learning Research
• Volume: 23
• Issue: 151
• Pages: 1-56
• Publication date: 2022-05-22
• EISSN: 1533-7928
• ISSN: 1532-4435
• Language: English
• Keywords: FFR