Improve model card: add pipeline tag, library, and links
Browse filesThis PR improves the model card for the model presented in [Embodied Instruction Following in Unknown Environments](https://huggingface.co/papers/2406.11818).
It enhances discoverability and provides more context by:
- Adding the `pipeline_tag: robotics`, ensuring the model appears under relevant searches at https://huggingface.co/models?pipeline_tag=robotics.
- Specifying `library_name: transformers` for better integration with the Hugging Face ecosystem, as the model uses components compatible with this library.
- Linking directly to the paper on the Hugging Face Hub.
- Adding links to the project page and the official GitHub repository for easy access to code and supplementary materials.
- Including the paper's abstract for a comprehensive overview of the work.
- Providing guidance on how to use/evaluate the model, referencing the original GitHub repository.
Please review and merge if everything looks good!
|
@@ -1,3 +1,50 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
pipeline_tag: robotics
|
| 4 |
+
library_name: transformers
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
# Embodied Instruction Following in Unknown Environments
|
| 8 |
+
|
| 9 |
+
This repository contains the official implementation and models for the paper:
|
| 10 |
+
[**Embodied Instruction Following in Unknown Environments**](https://huggingface.co/papers/2406.11818).
|
| 11 |
+
|
| 12 |
+
This model is a high-level task planner and low-level exploration controller, fine-tuned using LoRA on top of a Vicuna-v1-3-7B base model, primarily leveraged through the LLaVA framework.
|
| 13 |
+
|
| 14 |
+
## Abstract
|
| 15 |
+
Enabling embodied agents to complete complex human instructions from natural language is crucial to autonomous systems in household services. Conventional methods can only accomplish human instructions in the known environment where all interactive objects are provided to the embodied agent, and directly deploying the existing approaches for the unknown environment usually generates infeasible plans that manipulate non-existing objects. On the contrary, we propose an embodied instruction following (EIF) method for complex tasks in the unknown environment, where the agent efficiently explores the unknown environment to generate feasible plans with existing objects to accomplish abstract instructions. Specifically, we build a hierarchical embodied instruction following framework including the high-level task planner and the low-level exploration controller with multimodal large language models. We then construct a semantic representation map of the scene with dynamic region attention to demonstrate the known visual clues, where the goal of task planning and scene exploration is aligned for human instruction. For the task planner, we generate the feasible step-by-step plans for human goal accomplishment according to the task completion process and the known visual clues. For the exploration controller, the optimal navigation or object interaction policy is predicted based on the generated step-wise plans and the known visual clues. The experimental results demonstrate that our method can achieve 45.09% success rate in 204 complex human instructions such as making breakfast and tidying rooms in large house-level scenes.
|
| 16 |
+
|
| 17 |
+
## Links
|
| 18 |
+
* **Project Page**: https://gary3410.github.io/eif_unknown/
|
| 19 |
+
* **GitHub Repository**: https://github.com/gary3410/eif_unknown
|
| 20 |
+
|
| 21 |
+
## Usage and Evaluation
|
| 22 |
+
|
| 23 |
+
This model consists of LoRA adapters for a high-level planner and a low-level controller, built upon a Vicuna-v1-3-7B base model within the LLaVA framework. For detailed setup, data preparation, fine-tuning, and inference instructions, please refer to the comprehensive guide in the [official GitHub repository](https://github.com/gary3410/eif_unknown).
|
| 24 |
+
|
| 25 |
+
The models are typically used by loading the base model and then applying the LoRA weights. The project's GitHub repository provides scripts for evaluation.
|
| 26 |
+
|
| 27 |
+
Example evaluation command from the GitHub repository:
|
| 28 |
+
```bash
|
| 29 |
+
# Oracle setting
|
| 30 |
+
# Easy task
|
| 31 |
+
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m llava.serve.cli_llava_v3_nav_seg_gt \
|
| 32 |
+
--model-path ./checkpoints/llava-vicuna-v1-3-7b-finetune-frontier-lora-low-level-controller \
|
| 33 |
+
--model-path-s1 ./checkpoints/llava-vicuna-v1-3-7b-finetune-planner-lora-high-level-planner \
|
| 34 |
+
--model-base ./checkpoints/vicuna-v1-3-7b \
|
| 35 |
+
--image-file ./vision_dataset/llava_dataset_v8_easy_train/frontiers_feature \
|
| 36 |
+
--val-file ./data/spaced_parse_instruction_easy_v12_val.json
|
| 37 |
+
```
|
| 38 |
+
For other evaluation settings (e.g., Detic setting), please refer to the [GitHub repository](https://github.com/gary3410/eif_unknown).
|
| 39 |
+
|
| 40 |
+
## Citation
|
| 41 |
+
If you find this repository helpful, please consider citing:
|
| 42 |
+
|
| 43 |
+
```bibtex
|
| 44 |
+
@article{wu2024embodied,
|
| 45 |
+
title={Embodied instruction following in unknown environments},
|
| 46 |
+
author={Wu, Zhenyu and Wang, Ziwei and Xu, Xiuwei and Lu, Jiwen and Yan, Haibin},
|
| 47 |
+
journal={arXiv preprint arXiv:2406.11818},
|
| 48 |
+
year={2024}
|
| 49 |
+
}
|
| 50 |
+
```
|