Deep Sketched Output Kernel Regression for Structured Prediction

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ahmad, Tamim El, Yang, Junjie, Laforgue, Pierre, d'Alché-Buc, Florence
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909223334445056
author Ahmad, Tamim El
Yang, Junjie
Laforgue, Pierre
d'Alché-Buc, Florence
author_facet Ahmad, Tamim El
Yang, Junjie
Laforgue, Pierre
d'Alché-Buc, Florence
contents By leveraging the kernel trick in the output space, kernel-induced losses provide a principled way to define structured output prediction tasks for a wide variety of output modalities. In particular, they have been successfully used in the context of surrogate non-parametric regression, where the kernel trick is typically exploited in the input space as well. However, when inputs are images or texts, more expressive models such as deep neural networks seem more suited than non-parametric methods. In this work, we tackle the question of how to train neural networks to solve structured output prediction tasks, while still benefiting from the versatility and relevance of kernel-induced losses. We design a novel family of deep neural architectures, whose last layer predicts in a data-dependent finite-dimensional subspace of the infinite-dimensional output feature space deriving from the kernel-induced loss. This subspace is chosen as the span of the eigenfunctions of a randomly-approximated version of the empirical kernel covariance operator. Interestingly, this approach unlocks the use of gradient descent algorithms (and consequently of any neural architecture) for structured prediction. Experiments on synthetic tasks as well as real-world supervised graph prediction problems show the relevance of our method.
format Preprint
id arxiv_https___arxiv_org_abs_2406_09253
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deep Sketched Output Kernel Regression for Structured Prediction
Ahmad, Tamim El
Yang, Junjie
Laforgue, Pierre
d'Alché-Buc, Florence
Machine Learning
By leveraging the kernel trick in the output space, kernel-induced losses provide a principled way to define structured output prediction tasks for a wide variety of output modalities. In particular, they have been successfully used in the context of surrogate non-parametric regression, where the kernel trick is typically exploited in the input space as well. However, when inputs are images or texts, more expressive models such as deep neural networks seem more suited than non-parametric methods. In this work, we tackle the question of how to train neural networks to solve structured output prediction tasks, while still benefiting from the versatility and relevance of kernel-induced losses. We design a novel family of deep neural architectures, whose last layer predicts in a data-dependent finite-dimensional subspace of the infinite-dimensional output feature space deriving from the kernel-induced loss. This subspace is chosen as the span of the eigenfunctions of a randomly-approximated version of the empirical kernel covariance operator. Interestingly, this approach unlocks the use of gradient descent algorithms (and consequently of any neural architecture) for structured prediction. Experiments on synthetic tasks as well as real-world supervised graph prediction problems show the relevance of our method.
title Deep Sketched Output Kernel Regression for Structured Prediction
topic Machine Learning
url https://arxiv.org/abs/2406.09253