TGAS-ReID: Efficient architecture search for person re-identification via greedy decisions with topological order

Shengbo Chen,Kai Jiang,Xianrui Liu,Kangkang Yang,Zhou Lei
DOI: https://doi.org/10.1007/s10489-021-03097-5
IF: 5.3
2022-07-29
Applied Intelligence
Abstract:Person Re-Identification (Re-ID) technology is being developed rapidly due to the successful application of deep convolutional neural networks. However, the prevailing Re-ID models are usually built upon manually design backbones. In this paper, we propose using the TGAS-ReID which is automatically designed convolutional network backbones for Re-ID to substitute the backbones originally designed for classification such as ResNet and VGG. In the Re-ID tasks to search for a cell structure, greedy decisions are made instead of deriving the architecture after comprehensive training. In other words, at each decision epoch, according to the topological order, we first decide the candidates’ pool of the edges to progressively reduce the coupling of the internal nodes of the DAG. An edge is then selected based on edge importance, edge certainty, and selection stability. We then make a greedy optimal choice for the selected edge and prune the relevant parameters. To further improve the backbone’s representation capability of the features, we further introduce the triplet loss with batch hard mining as the retrieval loss. Extensive experiments demonstrate that the searched structure of the backbones reaches a performance level close to the previous work with a 20.8% shorter searching time. The proposed method also prevents the final CNNs network from suffering the well-known performance collapse by avoiding aggregation of the skip-connections.
computer science, artificial intelligence
What problem does this paper attempt to address?