Source-linked AI summary

Dexterous Grasp Transformer

Guo-Hao Xu, Yi-Lin Wei, Dian Zheng, Xiao-Ming Wu, Wei-Shi Zheng

arXiv:2404.18135v1cs.RO

TL;DR

Prior methods struggle to produce diverse, feasible grasps from the same object point cloud, often relying on repeated inference. DGTR uses transformer-based set prediction and progressive training and test-time adaptation; it reports high-quality, diverse grasp poses in one forward pass.

  • Problem

    Prior approaches struggle to generate feasible, diverse grasps from the same point cloud, with generative methods often producing nearly identical outputs and vanilla discriminative models predicting one pose.

    Source: §1 · p. 2
  • Method

    DGTR uses a transformer decoder with learnable grasp queries to predict a set of grasps, alongside dynamic-static matching training and adversarial-balanced test-time adaptation.

    Source: §1 · p. 3
  • Results

    DGTR achieves high-quality and diverse dexterous grasps with one forward pass.

    Source: §5 · p. 9
  • Takeaways & Limitations

    The authors see DGTR as having development potential for task-oriented and real-world dexterous grasp generation.

    Source: §5 · p. 9

Abstract

from arXiv · show

In this work, we propose a novel discriminative framework for dexterous grasp generation, named Dexterous Grasp TRansformer (DGTR), capable of predicting a diverse set of feasible grasp poses by processing the object point cloud with only one forward pass. We formulate dexterous grasp generation as a set prediction task and design a transformer-based grasping model for it. However, we identify that this set prediction paradigm encounters several optimization challenges in the field of dexterous grasping and results in restricted performance. To address these issues, we propose progressive strategies for both the training and testing phases. First, the dynamic-static matching training (DSMT) strategy is presented to enhance the optimization stability during the training phase. Second, we introduce the adversarial-balanced test-time adaptation (AB-TTA) with a pair of adversarial losses to improve grasping quality during the testing phase. Experimental results on the DexGraspNet dataset demonstrate the capability of DGTR to predict dexterous grasp poses with both high quality and diversity. Notably, while keeping high quality, the diversity of grasp poses predicted by DGTR significantly outperforms previous works in multiple metrics without any data pre-processing. Codes are available at https://github.com/iSEE-Laboratory/DGTR .

1. Introduction

Prior methods struggle to produce diverse, high-quality grasps from one object input without repeated inference. DGTR predicts a diverse grasp set in one pass and uses progressive training and test-time strategies to address its optimization challenge.

1. Introduction

  • Prior generative methods may produce nearly identical grasps for the same input, while vanilla discriminative models predict only one pose.Seeking diversity with either approach can require rotating the point cloud and inferring multiple times.Source: §1 · pp. 2–3, Figure 1 · p. 2
  • DGTR predicts a diverse set of feasible grasp poses from the complete object point cloud in one forward pass, without data preprocessing.The authors describe it as the first work to make this one-pass, diverse prediction.Source: §1 · p. 3
  • A large penetration-loss weight can collapse predictions to nearly identical grasps, whereas zero weight leads to severe object penetration.The authors identify instability in Hungarian matching, exacerbated by the penetration loss, as the main cause.Source: §1 · p. 3
  • DSMT trains first with dynamic matching without penetration loss, then with static matching and penetration loss; AB-TTA refines hand poses using opposing losses.One AB-TTA loss repels the hand from the object interior, while the other attracts it toward the surface; the strategies are reported to improve grasp quality and diversity.Source: §1 · p. 3, Figure 2 · p. 3
  • Experiments on DexGraspNet demonstrate DGTR’s capability to generate high-quality and diverse grasps.Figure 2 compares three grasps under a large penetration-loss weight, zero weight, and the progressive strategies.Source: §1 · p. 3, Figure 2 · p. 3

2. Related Works

Earlier dexterous-grasp methods include analytical, dataset-synthesis, generative, and discriminative approaches, but face limitations in speed, success rate, or diverse feasible predictions. DGTR applies set prediction with grasp-specific losses and progressive strategies to address challenges that conventional detection transformers do not handle.

2.1. Dexterous Grasp Generation

  • Existing methods struggle to generate feasible, diverse grasps from the same point cloud because conditioning or model architecture can restrict grasp direction.Analytical and dataset-synthesis approaches also face generation-speed and success-rate challenges; prior data-driven work includes generative models and a non-generative model that predicts one pose per instance.Source: §2.1 · p. 3
  • DGTR instead predicts a diverse set of grasps at one time with a transformer-based framework designed for dexterous grasp generation.Its overview uses a point-cloud encoder, transformer decoder, and feed-forward networks to predict grasp poses.Source: §2.1 · p. 3, Figure 3 · p. 4

2.2. Vision Transformer

  • Dexterous grasp generation from a complete point cloud is formulated as set prediction, but conventional detection transformers lack feasible-grasp supervision and face optimization challenges from grasp losses.DGTR equips the model with grasp losses and progressive strategies for stable training and penetration optimization.Source: §2.2 · p. 4

3. Dexterous Grasp Transformer

DGTR formulates dexterous grasp generation as set prediction, using a transformer to produce diverse grasp poses from an object point cloud. DSMT stabilizes training, while AB-TTA refines grasps at test time to balance contact and penetration.

3.1. Problem Formulation

  • DGTR predicts a set of grasp poses from the complete object point cloud, representing each pose with rotation, translation, and hand joint angles.The formulation targets high-quality, diverse poses; the architecture uses a point-cloud encoder, transformer decoder, and prediction heads.Source: §3.1 · p. 4, §3.2 · p. 4

3.2. DGTR Architecture

  • Hungarian matching can destabilize optimization: strong penetration penalties encourage nearly identical grasps, while weak or absent penalties permit severe object penetration.The penetration loss increases matching instability, creating ambiguous optimization targets across queries.Source: §3.2 · p. 4, §3.3 · p. 5

3.3. Dynamic-Static Matching Training Strategy

  • DSMT trains first with dynamic matching, then fixes those assignments before adding penetration and hand-object distance losses.Dynamic training learns diverse grasping patterns; static matching warm-up adapts the model to fixed assignments before static penetration training.Source: §3.3 · p. 5

3.4. Adversarial-Balanced Test-Time Adaptation

  • AB-TTA refines grasps at test time by balancing object-penetration and hand-object-distance losses, with translation moderation and a generalized distance loss.The generalized loss keeps initially contacting keypoints in contact; moderated root translation limits over-optimization of penetration.Source: §3.4 · pp. 5–6

3.5. Grasp Losses

  • DGTR training combines hand-parameter regression, hand-shape Chamfer distance, object and self-penetration penalties, and Hungarian-matched prediction-to-ground-truth losses.Regression covers translation, joint angles, and rotation; when prediction and target counts differ, redundant items remain unmatched.Source: §3.5 · pp. 6–7

4.1. Dataset and Evaluation Metrics

The evaluation uses DexGraspNet’s official training-validation split and measures both grasp quality and grasp diversity. Quality metrics cover stability, penetration, torque balance, and success; diversity metrics measure occupied translations, rotations, and joint angles.

  • DexGraspNet contains 1.32 million ShadowHand grasps across 5,355 objects from more than 133 categories, and experiments use its official training-validation split.Source: §4.1 · p. 7
  • Five quality metrics assess stability, maximal penetration, non-penetration, torque balance, and success under six possible gravity directions.The evaluation defines a valid grasp as one that holds the object steadily under any one of the six gravity directions.Source: §4.1 · p. 7
  • Three diversity metrics measure occupied translation, rotation, and joint-angle bins, with higher values indicating broader spatial, orientational, or gesture coverage.The continuous parameter spaces are discretized into 16 uniform bins; translation directions use Fibonacci sampling on a unit sphere.Source: §4.1 · p. 7

4.2. Implementation Details

DGTR uses 16 grasp queries and a staged DSMT training schedule, with specified loss weights and AB-TTA settings. The reported tables cover one-pass comparisons, multiple-pass comparisons, and component effectiveness.

  • DGTR uses 16 queries and trains DSMT stages for 15, 5, and 5 epochs on a single RTX 4090 GPU.Source: §4.2 · p. 7
  • The implementation sets Hungarian costs to 2.0, 1.0, and 2.0, varies λ6 between 0.0 and 50.0 across training stages, and uses βt = 0, α1 = 5, α2 = 3, and α3 = 5 for AB-TTA.The other listed loss weights are λ1 = 10.0, λ2 = 10.0, λ3 = 10.0, λ4 = 1.0, and λ5 = 10.0; SMPT uses distance loss weight 10.0.Source: §4.2 · p. 7
  • The reported tables cover one-forward-pass state-of-the-art results, multiple-pass inference time and grasp counts, and DGTR component effectiveness.Table 1 identifies DGTR* as a quality-focused variant and DDG as a quality reference outside the same setting; Table 2 defines total inference time, pass count, and grasps per pass.Source: Table 1 · p. 8, Table 2 · p. 8, Table 3 · p. 8

4.3. Dexterous Grasp Generation Performance

DGTR is evaluated against grasp-generation methods in both one-pass and multiple-pass settings. The reported results describe strong quality and diversity in one pass, with lower inference time and greater translation and rotation diversity than 16-pass UniDexGrasp.

4.3.1 Comparison with SOTA in one forward pass

  • DGTR has more balanced quality and a higher success rate than generative models, while quality-focused DGTR* is comparable to DDG.DGTR* selects its top four results by contact points and object penetration without extra inputs; DDG is a quality reference rather than a same-setting comparison.Source: §4.3.1 · p. 8
  • DGTR exceeds UniDexGrasp and GraspTTA by a large gap in δt and δr, while SceneDiffuser has higher diversity but much lower quality.The paper’s qualitative examples also show DGTR generating grasps from varied directions in one forward pass.Source: §4.3.1 · p. 8, Figure 7 · p. 8
  • DGTR achieves overall state-of-the-art performance for high-quality and diverse grasps in the one-forward-pass comparison.Source: §4.3.1 · p. 8

4.3.2 Comparison with SOTA in multiple forward pass

  • Against UniDexGrasp with 16 forward passes, DGTR uses significantly less time and achieves higher δt and δr.UniDexGrasp rotates the object point cloud and passes it to the model multiple times; DGTR’s results indicate broader hand-position and direction coverage.Source: §4.3.2 · p. 8

4.4. Ablation Study

The ablations report substantial gains from both DSMT and AB-TTA. Stage comparisons highlight static matching’s role in reducing penetration, while AB-TTA’s generalized distance loss and translation moderation are reported as key designs.

4.4.1 Dynamic-Static Matching Training Strategy

  • DSMT increases Q1 by 3.5 times and reduces Pen. by nearly 50%.Source: §4.4.1 · p. 8
  • The DSMT stage ablation highlights static matching as critical for directing optimization and reducing object penetration.Table 4 compares DMT, SMW, and SMPT stages and identifies static matching and warm-up for SMPT.Source: §4.4.1 · p. 8, Table 4 · p. 9

4.4.2 Adversarial-Balanced Test-Time Adaptation

  • Generalized TTA-distance loss and translation moderation are reported as indispensable; simple penetration and vanilla-distance TTA has limited effect.AB-TTA also outperforms ContactNet-TTA in grasp quality and can boost its Q1 and ηtb.Source: Table 7 · p. 9, §4.4.2 · p. 9
  • AB-TTA increases Q1 by 1.85-fold and improves both ηnp and ηtb.Source: §4.4.2 · p. 9

4.5. DGTR Analysis

The analysis examines how penetration-loss weighting and grasp-query count affect DGTR. Increasing the penetration weight reduces penetration but brings a non-contact issue, while more queries increase diversity and challenge grasp quality.

4.5.1 Loss Weight for Object Penetration

  • Increasing λpen reduces object penetration but concurrently causes severe non-contact, and gradually raising it from 0 to 50 after warm-up does not solve the problem.The authors attribute model collapse under a large penetration loss to unstable Hungarian matching; DSMT and AB-TTA tackle the challenge effectively.Source: §4.5.1 · p. 9, Table 6 · p. 9

4.5.2 Number of Grasping Queries

  • As the number of grasping queries increases, grasp quality Q1 tends to decrease while diversity increases.The authors interpret this pattern as evidence that simultaneously learning a larger set of grasp poses is challenging.Source: §4.5.2 · p. 9

5. Conclusions

DGTR is a discriminative framework for dexterous grasp generation, with DSMT and AB-TTA as progressive training and testing strategies. The authors report high-quality, diverse grasp poses in one forward pass and identify task-oriented and real-world generation as potential directions.

5. Conclusions

  • DGTR achieves both high grasp quality and diversity while predicting grasps with one forward pass.The authors describe it as the first work to introduce set prediction into the dexterous grasp domain.Source: §5 · p. 9
  • DSMT and AB-TTA substantially improve grasping stability and reduce penetration.The strategies apply during training and testing, respectively.Source: §5 · p. 9
  • The authors identify task-oriented and real-world dexterous grasp generation as potential development directions for DGTR.Source: §5 · p. 9
Loading 2404.18135v1…