Perception Matters: Enhancing Embodied AI with Uncertainty-Aware Semantic Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Prasanna, Sai, Honerkamp, Daniel, Sirohi, Kshitij, Welschehold, Tim, Burgard, Wolfram, Valada, Abhinav
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916566079111168
author Prasanna, Sai
Honerkamp, Daniel
Sirohi, Kshitij
Welschehold, Tim
Burgard, Wolfram
Valada, Abhinav
author_facet Prasanna, Sai
Honerkamp, Daniel
Sirohi, Kshitij
Welschehold, Tim
Burgard, Wolfram
Valada, Abhinav
contents Embodied AI has made significant progress acting in unexplored environments. However, tasks such as object search have largely focused on efficient policy learning. In this work, we identify several gaps in current search methods: They largely focus on dated perception models, neglect temporal aggregation, and transfer from ground truth directly to noisy perception at test time, without accounting for the resulting overconfidence in the perceived state. We address the identified problems through calibrated perception probabilities and uncertainty across aggregation and found decisions, thereby adapting the models for sequential tasks. The resulting methods can be directly integrated with pretrained models across a wide family of existing search approaches at no additional training cost. We perform extensive evaluations of aggregation methods across both different semantic perception models and policies, confirming the importance of calibrated uncertainties in both the aggregation and found decisions. We make the code and trained models available at https://semantic-search.cs.uni-freiburg.de.
format Preprint
id arxiv_https___arxiv_org_abs_2408_02297
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Perception Matters: Enhancing Embodied AI with Uncertainty-Aware Semantic Segmentation
Prasanna, Sai
Honerkamp, Daniel
Sirohi, Kshitij
Welschehold, Tim
Burgard, Wolfram
Valada, Abhinav
Robotics
Computer Vision and Pattern Recognition
Embodied AI has made significant progress acting in unexplored environments. However, tasks such as object search have largely focused on efficient policy learning. In this work, we identify several gaps in current search methods: They largely focus on dated perception models, neglect temporal aggregation, and transfer from ground truth directly to noisy perception at test time, without accounting for the resulting overconfidence in the perceived state. We address the identified problems through calibrated perception probabilities and uncertainty across aggregation and found decisions, thereby adapting the models for sequential tasks. The resulting methods can be directly integrated with pretrained models across a wide family of existing search approaches at no additional training cost. We perform extensive evaluations of aggregation methods across both different semantic perception models and policies, confirming the importance of calibrated uncertainties in both the aggregation and found decisions. We make the code and trained models available at https://semantic-search.cs.uni-freiburg.de.
title Perception Matters: Enhancing Embodied AI with Uncertainty-Aware Semantic Segmentation
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.02297