Advancing ALS Applications with Large-Scale Pre-training: Dataset Development and Downstream Assessment
Haoyi Xiu, Xin Liu, Taehoon Kim, Kyoung-Sook Kim
TL;DR
This work tackles the lack of large-scale pre-training data for airborne LiDAR (ALS) by constructing a continental-scale ALS tile dataset from USGS 3DEP using a geospatial sampling strategy guided by NLCD land cover and DEM-derived slope. The authors adopt BEV-MAE, a masked autoencoder for 3D outdoor point clouds, to pre-train on this dataset and fine-tune on tree species classification, terrain scene recognition, and point cloud semantic segmentation. Results show that pre-training yields consistent improvements over scratch baselines for tree species and terrain tasks, with dataset scale and sampling strategy driving performance gains, while urban segmentation benefits are more modest. The study demonstrates the transferability of representations learned from the proposed ALS dataset and highlights directions for enhancing reconstruction richness and developing ALS-specific SSL methods, with code and models released to the community.
Abstract
The pre-training and fine-tuning paradigm has revolutionized satellite remote sensing applications. However, this approach remains largely underexplored for airborne laser scanning (ALS), an important technology for applications such as forest management and urban planning. In this study, we address this gap by constructing a large-scale ALS point cloud dataset and evaluating its impact on downstream applications. Our dataset comprises ALS point clouds collected across the contiguous United States, provided by the United States Geological Survey's 3D Elevation Program. To ensure efficient data collection while capturing diverse land cover and terrain types, we introduce a geospatial sampling method that selects point cloud tiles based on land cover maps and digital elevation models. As a baseline self-supervised learning model, we adopt BEV-MAE, a state-of-the-art masked autoencoder for 3D outdoor point clouds, and pre-train it on the constructed dataset. The pre-trained models are subsequently fine-tuned for downstream tasks, including tree species classification, terrain scene recognition, and point cloud semantic segmentation. Our results show that the pre-trained models significantly outperform their scratch counterparts across all downstream tasks, demonstrating the transferability of the representations learned from the proposed dataset. Furthermore, we observe that scaling the dataset using our geospatial sampling method consistently enhances performance, whereas pre-training on datasets constructed with random sampling fails to achieve similar improvements. These findings highlight the utility of the constructed dataset and the effectiveness of our sampling strategy in the pre-training and fine-tuning paradigm. The source code and pre-trained models will be made publicly available at \url{https://github.com/martianxiu/ALS_pretraining}.
