kth.sePublikationer KTH
Ändra sökning
Länk till posten
Permanent länk

Direktlänk
Tegner, Gustaf
Publikationer (3 of 3) Visa alla publikationer
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
Öppna denna publikation i ny flik eller fönster >>Reducing Variance in Meta-Learning via Laplace Approximation for Regression Tasks
Visa övriga...
2024 (Engelska)Ingår i: Transactions on Machine Learning Research, E-ISSN 2835-8856, Vol. 2024Artikel i tidskrift (Refereegranskat) 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.

Ort, förlag, år, upplaga, sidor
Transactions on Machine Learning Research, 2024
Nationell ämneskategori
Robotik och automation Reglerteknik
Identifikatorer
urn:nbn:se:kth:diva-361197 (URN)2-s2.0-85219566964 (Scopus ID)
Anmärkning

QC 20250312

Tillgänglig från: 2025-03-12 Skapad: 2025-03-12 Senast uppdaterad: 2026-02-16Bibliografiskt granskad
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
Öppna denna publikation i ny flik eller fönster >>Equivariant Representation Learning via Class-Pose Decomposition
2023 (Engelska)Ingår i: Proceedings of the 26th International Conference on Artificial Intelligence and Statistics, AISTATS 2023, ML Research Press , 2023, Vol. 206, s. 4745-4756Konferensbidrag, Publicerat paper (Refereegranskat)
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.

Ort, förlag, år, upplaga, sidor
ML Research Press, 2023
Nationell ämneskategori
Robotik och automation
Identifikatorer
urn:nbn:se:kth:diva-334435 (URN)001222727704045 ()2-s2.0-85165155542 (Scopus ID)
Konferens
26th International Conference on Artificial Intelligence and Statistics, AISTATS 2023, Valencia, Spain, Apr 25 2023 - Apr 27 2023
Anmärkning

QC 20241204

Tillgänglig från: 2023-08-21 Skapad: 2023-08-21 Senast uppdaterad: 2025-02-09Bibliografiskt granskad
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
Öppna denna publikation i ny flik eller fönster >>Learning Coarsened Dynamic Graph Representations for Deformable Object Manipulation
Visa övriga...
2021 (Engelska)Ingår i: 2021 20Th International Conference On Advanced Robotics (ICAR), IEEE , 2021, s. 955-960Konferensbidrag, Publicerat paper (Refereegranskat)
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.

Ort, förlag, år, upplaga, sidor
IEEE, 2021
Nationell ämneskategori
Robotik och automation Datorgrafik och datorseende
Identifikatorer
urn:nbn:se:kth:diva-310888 (URN)10.1109/ICAR53236.2021.9659451 (DOI)000766318900144 ()2-s2.0-85124707484 (Scopus ID)
Konferens
20th International Conference on Advanced Robotics (ICAR), DEC 07-10, 2021, ELECTR NETWORK
Anmärkning

QC 20220412

Tillgänglig från: 2022-04-12 Skapad: 2022-04-12 Senast uppdaterad: 2025-02-05Bibliografiskt granskad
Organisationer

Sök vidare i DiVA

Visa alla publikationer