Abstract:Multi-person pose estimation has achieved great progress in recent years, even though, the precise prediction for occluded and invisible hard keypoints remains challenging. Most of the human pose estimation networks are equipped with an image classification-based pose encoder for feature extraction and a handcrafted pose decoder for high-resolution representations. However, the pose encoder might be sub-optimal because of the gap between image classification and pose estimation. The widely used multi-scale feature fusion in pose decoder is still coarse and cannot provide sufficient high-resolution details for hard keypoints. Neural Architecture Search (NAS) has shown great potential in many visual tasks to automatically search efficient networks. In this work, we present the Pose-native Network Architecture Search (PoseNAS) to simultaneously design a better pose encoder and pose decoder for pose estimation. Specifically, we directly search a data-oriented pose encoder with stacked searchable cells, which can provide an optimum feature extractor for the pose specific task. In the pose decoder, we exploit scale-adaptive fusion cells to promote rich information exchange across the multi-scale feature maps. Meanwhile, the pose decoder adopts a Fusion-and-Enhancement manner to progressively boost the high-resolution representations that are non-trivial for the precious prediction of hard keypoints. With the exquisitely designed search space and search strategy, PoseNAS can simultaneously search all modules in an end-to-end manner. PoseNAS achieves state-of-the-art performance on three public datasets, MPII, COCO, and PoseTrack, with small-scale parameters compared with the existing methods. Our best model obtains 76.7% mAP and 75.9% mAP on the COCO validation set and test set with only 33.6M parameters. Code and implementation are available at https://github.com/for-code0216/PoseNAS.

Searching part-specific neural fabrics for human pose estimation

Pose Neural Fabrics Search

Pose-native Network Architecture Search for Multi-person Human Pose Estimation

Learning a Robust Part-Aware Monocular 3D Human Pose Estimator via Neural Architecture Search

ZoomNAS: Searching for Whole-Body Human Pose Estimation in the Wild

Prioritized Architecture Sampling with Monto-Carlo Tree Search

Toward Less Constrained Macro-Neural Architecture Search

Towards Less Constrained Macro-Neural Architecture Search

FBNetV5: Neural Architecture Search for Multiple Tasks in One Run

Manas: multi-agent neural architecture search

ViPNAS: Efficient Video Pose Estimation via Neural Architecture Search

HR-NAS: Searching Efficient High-Resolution Neural Architectures with Lightweight Transformers

DCNAS: Densely Connected Neural Architecture Search for Semantic Image Segmentation

Large Scale Neural Architecture Search with Polyharmonic Splines

Neural Architecture Search Via Combinatorial Multi-Armed Bandit.

SM-NAS: Structural-to-Modular Neural Architecture Search for Object Detection

einspace: Searching for Neural Architectures from Fundamental Operations

AtomNAS: Fine-Grained End-to-End Neural Architecture Search

C2FNAS: Coarse-to-Fine Neural Architecture Search for 3D Medical Image Segmentation

AutoSpace: Neural Architecture Search with Less Human Interference

MedNAS: Multi-Scale Training-Free Neural Architecture Search for Medical Image Analysis