Mars Terrain Segmentation with Less Labels
- URL: http://arxiv.org/abs/2202.00791v1
- Date: Tue, 1 Feb 2022 22:25:15 GMT
- Title: Mars Terrain Segmentation with Less Labels
- Authors: Edwin Goh, Jingdao Chen, Brian Wilson
- Abstract summary: This research proposes a semi-supervised learning framework for Mars terrain segmentation.
It incorporates a backbone module which is trained using a contrastive loss function and an output atrous convolution module.
The proposed model is able to achieve a segmentation accuracy of 91.1% using only 161 training images.
- Score: 1.1745324895296465
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Planetary rover systems need to perform terrain segmentation to identify
drivable areas as well as identify specific types of soil for sample
collection. The latest Martian terrain segmentation methods rely on supervised
learning which is very data hungry and difficult to train where only a small
number of labeled samples are available. Moreover, the semantic classes are
defined differently for different applications (e.g., rover traversal vs.
geological) and as a result the network has to be trained from scratch each
time, which is an inefficient use of resources. This research proposes a
semi-supervised learning framework for Mars terrain segmentation where a deep
segmentation network trained in an unsupervised manner on unlabeled images is
transferred to the task of terrain segmentation trained on few labeled images.
The network incorporates a backbone module which is trained using a contrastive
loss function and an output atrous convolution module which is trained using a
pixel-wise cross-entropy loss function. Evaluation results using the metric of
segmentation accuracy show that the proposed method with contrastive
pretraining outperforms plain supervised learning by 2%-10%. Moreover, the
proposed model is able to achieve a segmentation accuracy of 91.1% using only
161 training images (1% of the original dataset) compared to 81.9% with plain
supervised learning.
Related papers
- Weakly-supervised Camera Localization by Ground-to-satellite Image Registration [52.54992898069471]
We propose a weakly supervised learning strategy for ground-to-satellite image registration.
It derives positive and negative satellite images for each ground image.
We also propose a self-supervision strategy for cross-view image relative rotation estimation.
arXiv Detail & Related papers (2024-09-10T12:57:16Z) - Cross-Level Distillation and Feature Denoising for Cross-Domain Few-Shot
Classification [49.36348058247138]
We tackle the problem of cross-domain few-shot classification by making a small proportion of unlabeled images in the target domain accessible in the training stage.
We meticulously design a cross-level knowledge distillation method, which can strengthen the ability of the model to extract more discriminative features in the target dataset.
Our approach can surpass the previous state-of-the-art method, Dynamic-Distillation, by 5.44% on 1-shot and 1.37% on 5-shot classification tasks.
arXiv Detail & Related papers (2023-11-04T12:28:04Z) - GenCo: An Auxiliary Generator from Contrastive Learning for Enhanced
Few-Shot Learning in Remote Sensing [9.504503675097137]
We introduce a generator-based contrastive learning framework (GenCo) that pre-trains backbones and simultaneously explores variants of feature samples.
In fine-tuning, the auxiliary generator can be used to enrich limited labeled data samples in feature space.
We demonstrate the effectiveness of our method in improving few-shot learning performance on two key remote sensing datasets.
arXiv Detail & Related papers (2023-07-27T03:59:19Z) - Semi-Supervised Domain Adaptation for Semantic Segmentation of Roads
from Satellite Images [1.1470070927586016]
This paper presents the preliminary findings of a semi-supervised segmentation method for extracting roads from sattelite images.
A semi-supervised field adaptation method based on pseudo-labeling and Minimum Class Confusion has been proposed.
arXiv Detail & Related papers (2022-12-26T10:50:40Z) - Mixed-domain Training Improves Multi-Mission Terrain Segmentation [0.9566312408744931]
Current Martian terrain segmentation models require retraining for deployment across different domains.
This research proposes a semi-supervised learning approach that leverages unsupervised contrastive pretraining of a backbone for a multi-mission semantic segmentation for Martian surfaces.
arXiv Detail & Related papers (2022-09-27T20:25:24Z) - Semi-supervised domain adaptation with CycleGAN guided by a downstream
task loss [4.941630596191806]
Domain adaptation is of huge interest as labeling is an expensive and error-prone task.
Image-to-image approaches can be used to mitigate the shift in the input.
We propose a "task aware" version of a GAN in an image-to-image domain adaptation approach.
arXiv Detail & Related papers (2022-08-18T13:13:30Z) - PCA: Semi-supervised Segmentation with Patch Confidence Adversarial
Training [52.895952593202054]
We propose a new semi-supervised adversarial method called Patch Confidence Adrial Training (PCA) for medical image segmentation.
PCA learns the pixel structure and context information in each patch to get enough gradient feedback, which aids the discriminator in convergent to an optimal state.
Our method outperforms the state-of-the-art semi-supervised methods, which demonstrates its effectiveness for medical image segmentation.
arXiv Detail & Related papers (2022-07-24T07:45:47Z) - Segmentation of VHR EO Images using Unsupervised Learning [19.00071868539993]
We propose an unsupervised semantic segmentation method that can be trained using just a single unlabeled scene.
The proposed method exploits this property to sample smaller patches from the larger scene.
After unsupervised training on the target image/scene, the model automatically segregates the major classes present in the scene and produces the segmentation map.
arXiv Detail & Related papers (2021-07-09T11:42:48Z) - Seasonal Contrast: Unsupervised Pre-Training from Uncurated Remote
Sensing Data [64.40187171234838]
Seasonal Contrast (SeCo) is an effective pipeline to leverage unlabeled data for in-domain pre-training of re-mote sensing representations.
SeCo will be made public to facilitate transfer learning and enable rapid progress in re-mote sensing applications.
arXiv Detail & Related papers (2021-03-30T18:26:39Z) - Region Comparison Network for Interpretable Few-shot Image
Classification [97.97902360117368]
Few-shot image classification has been proposed to effectively use only a limited number of labeled examples to train models for new classes.
We propose a metric learning based method named Region Comparison Network (RCN), which is able to reveal how few-shot learning works.
We also present a new way to generalize the interpretability from the level of tasks to categories.
arXiv Detail & Related papers (2020-09-08T07:29:05Z) - Weakly Supervised Deep Nuclei Segmentation Using Partial Points
Annotation in Histopathology Images [51.893494939675314]
We propose a novel weakly supervised segmentation framework based on partial points annotation.
We show that our method can achieve competitive performance compared to the fully supervised counterpart and the state-of-the-art methods.
arXiv Detail & Related papers (2020-07-10T15:41:29Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.