← Search

Yuhei Umeda

9 accepted papers

2025

Instance-Optimal Pure Exploration for Linear Bandits on Continuous Arms

ICML 2025poster

This paper studies a pure exploration problem with linear bandit feedback on continuous arm sets, aiming to identify an $\epsilon$-optimal arm with high probability. Previous approaches for continuous arm sets have employed instance-independent methods due to technical challenges such as the infinit…

Cited by 0SourcePDFScholar
2024

Selective Mixup Fine-Tuning for Optimizing Non-Decomposable Objectives

ICLR 2024spotlight

The rise in internet usage has led to the generation of massive amounts of data, resulting in the adoption of various supervised and semi-supervised machine learning algorithms, which can effectively utilize the colossal amount of data to train models. However, before deploying these models in the r…

2022

Cost-Sensitive Self-Training for Optimizing Non-Decomposable Metrics

NeurIPS 2022accept

Self-training based semi-supervised learning algorithms have enabled the learning of highly accurate deep neural networks, using only a fraction of labeled data. However, the majority of work on self-training has focused on the objective of improving accuracy whereas practical machine learning syste…

2021

ATOL: Measure Vectorization for Automatic Topologically-Oriented Learning

AISTATS 2021poster

Robust topological information commonly comes in the form of a set of persistence diagrams, finite measures that are in nature uneasy to affix to generic machine learning frameworks. We introduce a fast, learnt, unsupervised vectorization method for measures in Euclidean spaces and use it for reflec…

Cited by 29SourcePDFScholar
2021

Optimizing persistent homology based functions

ICML 2021oral

Solving optimization tasks based on functions and losses with a topological flavor is a very active and growing field of research in data science and Topological Data Analysis, with applications in non-convex optimization, statistics and machine learning. However, the approaches proposed in the lite…

2021

Topological Uncertainty: Monitoring Trained Neural Networks through Persistence of Activation Graphs

IJCAI 2021poster

Although neural networks are capable of reaching astonishing performance on a wide variety of contexts, properly training networks on complicated tasks requires expertise and can be expensive from a computational perspective. In industrial applications, data coming from an open-world setting might w…

Cited by 28SourcePDFScholar
2020

PersLay: A Neural Network Layer for Persistence Diagrams and New Graph Topological Signatures

AISTATS 2020poster

Persistence diagrams, the most common descriptors of Topological Data Analysis, encode topological properties of data and have already proved pivotal in many different applications of data science. However, since the metric space of persistence diagrams is not Hilbert, they end up being difficult in…