kth.sePublications KTH
Change search
Link to record
Permanent link

Direct link
Tegner, Gustaf
Publications (3 of 3) Show all publications
Reichlin, A., Tegner, G., Vasco, M., Yin, H., Björkman, M. & Kragic, D. (2024). Reducing Variance in Meta-Learning via Laplace Approximation for Regression Tasks. Transactions on Machine Learning Research, 2024
Open this publication in new window or tab >>Reducing Variance in Meta-Learning via Laplace Approximation for Regression Tasks
Show others...
2024 (English)In: Transactions on Machine Learning Research, E-ISSN 2835-8856, Vol. 2024Article in journal (Refereed) Published
Abstract [en]

Given a finite set of sample points, meta-learning algorithms aim to learn an optimal adaptation strategy for new, unseen tasks. Often, this data can be ambiguous as it might belong to different tasks concurrently. This is particularly the case in meta-regression tasks. In such cases, the estimated adaptation strategy is subject to high variance due to the limited amount of support data for each task, which often leads to sub-optimal generalization performance. In this work, we address the problem of variance reduction in gradient-based meta-learning and formalize the class of problems prone to this, a condition we refer to as task overlap. Specifically, we propose a novel approach that reduces the variance of the gradient estimate by weighing each support point individually by the variance of its posterior over the parameters. To estimate the posterior, we utilize the Laplace approximation, which allows us to express the variance in terms of the curvature of the loss landscape of our meta-learner. Experimental results demonstrate the effectiveness of the proposed method and highlight the importance of variance reduction in meta-learning.

Place, publisher, year, edition, pages
Transactions on Machine Learning Research, 2024
National Category
Robotics and automation Control Engineering
Identifiers
urn:nbn:se:kth:diva-361197 (URN)2-s2.0-85219566964 (Scopus ID)
Note

QC 20250312

Available from: 2025-03-12 Created: 2025-03-12 Last updated: 2026-02-16Bibliographically approved
Marchetti, G. L., Tegner, G., Varava, A. & Kragic, D. (2023). Equivariant Representation Learning via Class-Pose Decomposition. In: Proceedings of the 26th International Conference on Artificial Intelligence and Statistics, AISTATS 2023: . Paper presented at 26th International Conference on Artificial Intelligence and Statistics, AISTATS 2023, Valencia, Spain, Apr 25 2023 - Apr 27 2023 (pp. 4745-4756). ML Research Press, 206
Open this publication in new window or tab >>Equivariant Representation Learning via Class-Pose Decomposition
2023 (English)In: Proceedings of the 26th International Conference on Artificial Intelligence and Statistics, AISTATS 2023, ML Research Press , 2023, Vol. 206, p. 4745-4756Conference paper, Published paper (Refereed)
Abstract [en]

We introduce a general method for learning representations that are equivariant to symmetries of data. Our central idea is to decompose the latent space into an invariant factor and the symmetry group itself. The components semantically correspond to intrinsic data classes and poses respectively. The learner is trained on a loss encouraging equivariance based on supervision from relative symmetry information. The approach is motivated by theoretical results from group theory and guarantees representations that are lossless, interpretable and disentangled. We provide an empirical investigation via experiments involving datasets with a variety of symmetries. Results show that our representations capture the geometry of data and outperform other equivariant representation learning frameworks.

Place, publisher, year, edition, pages
ML Research Press, 2023
National Category
Robotics and automation
Identifiers
urn:nbn:se:kth:diva-334435 (URN)001222727704045 ()2-s2.0-85165155542 (Scopus ID)
Conference
26th International Conference on Artificial Intelligence and Statistics, AISTATS 2023, Valencia, Spain, Apr 25 2023 - Apr 27 2023
Note

QC 20241204

Available from: 2023-08-21 Created: 2023-08-21 Last updated: 2025-02-09Bibliographically approved
Marchetti, G. L., Moletta, M., Tegner, G., Shi, P., Varava, A., Kravchenko, O. & Kragic, D. (2021). Learning Coarsened Dynamic Graph Representations for Deformable Object Manipulation. In: 2021 20Th International Conference On Advanced Robotics (ICAR): . Paper presented at 20th International Conference on Advanced Robotics (ICAR), DEC 07-10, 2021, ELECTR NETWORK (pp. 955-960). IEEE
Open this publication in new window or tab >>Learning Coarsened Dynamic Graph Representations for Deformable Object Manipulation
Show others...
2021 (English)In: 2021 20Th International Conference On Advanced Robotics (ICAR), IEEE , 2021, p. 955-960Conference paper, Published paper (Refereed)
Abstract [en]

Manipulation of deformable objects has long been a challenging task in robotics. Their high-dimensional configuration space and complex dynamics make them difficult to consider for tasks such as robotic manipulation. In this paper, we address the problem of learning efficient representations of deformable objects which lend themselves better suitable for downstream robotics tasks. In particular, we consider graph-based representations of deformable objects which arise naturally from their point-cloud representation. Through manipulation, we learn to coarsen this graph into a simpler representation which still captures the necessary dynamics of the object. Our model consists of (a) a Cluster Assignment Model which takes the initial graph and coarsens it, (b) a Coarsened Dynamics Model that approximates the dynamics of the coarsened graph and (c) a Forward Prediction Model which predicts the next state of the ground truth graph. After end-to-end training, the Cluster Assignment Model learns to build coarse representations which better capture the dynamics compared to conventional clustering methods such as K-means. We evaluate our method on three sets of experiments: rigid objects, rigid objects with pair-wise interactions and a simulated dataset of a shirt.

Place, publisher, year, edition, pages
IEEE, 2021
National Category
Robotics and automation Computer graphics and computer vision
Identifiers
urn:nbn:se:kth:diva-310888 (URN)10.1109/ICAR53236.2021.9659451 (DOI)000766318900144 ()2-s2.0-85124707484 (Scopus ID)
Conference
20th International Conference on Advanced Robotics (ICAR), DEC 07-10, 2021, ELECTR NETWORK
Note

QC 20220412

Available from: 2022-04-12 Created: 2022-04-12 Last updated: 2025-02-05Bibliographically approved
Organisations

Search in DiVA

Show all publications