正在加载视频...

视频加载失败

Introducing Masked Trajectory Modeling (MTM), a new general-purpose framework for sequential decision making. A single transformer trained with MTM can exhibit multiple capabilities by simply choosing different masking patterns at inference time. Accepted at ICML 2023. 🧵👇

93,121 次观看 • 3 年前 •via X (Twitter)

11 条评论

Philipp Wu 的头像
Philipp Wu3 年前

MTM trains by reconstructing a masked input sequence using random autoregressive masking (think BERT with a prediction prior). We prompt the model with different masking patterns to achieve specific capabilities such as behavior cloning and inverse dynamics.

Philipp Wu 的头像
Philipp Wu3 年前

We also evaluate MTM in offline RL settings, such as D4RL. At test time we prompt the model with a desired return. We find that MTM is comparable to prior work while also also having the capability to be used for other tasks.

Philipp Wu 的头像
Philipp Wu3 年前

MTM is able to train on datasets with missing data modes. Ex: we might have a large dataset of state trajectories, but only a small amount with action labels. MTM is able to train on both without any change to the algorithm, simply by treating missing data modes as a forced mask.

Philipp Wu 的头像
Philipp Wu3 年前

We test if MTM is able to leverage the trajectories without actions. Here we compare various methods that only train on the labeled subset and our Heteromodal MTM which also trains on trajectories without action labels. Heteromodal MTM is able to improve task performance.

Philipp Wu 的头像
Philipp Wu3 年前

MTM can also be used to learn representations. After pretraining on generic trajectories, we use TD3 on top of the learned MTM representations. We find that these representations help with faster learning and in some cases, improve on asymptotic performance.

Philipp Wu 的头像
Philipp Wu3 年前

If you are interested in this line of work, also checkout MaskDP ( from my lab mate @fangchenliu. Our work extends on some of the ideas presented here and explores a different set of capabilities and features of this paradigm.

Philipp Wu 的头像
Philipp Wu3 年前

If you want to get a quick start with MTM, checkout this notebook which illustrates how to train and use MTM on a simple sinusoidal dataset.

Philipp Wu 的头像
Philipp Wu3 年前

Thanks for the help and assistance of amazing collaborators! @arjunmajum, @kevinleestone, @yixin_lin_, @IMordatch, @pabbeel, @aravindr93 If you are interested in more details checkout the links below! Website: Code:

Philipp Wu 的头像
Philipp Wu3 年前

Paper now out on arxiv!

kourosh hakhamaneshi 的头像
kourosh hakhamaneshi3 年前

Nice. It finally worked. Congrats. 🎉

Philipp Wu 的头像
Philipp Wu3 年前

😂 thanks kourosh!!

相关视频

In this demo we extend our prior work on obstacles avoidance at aggressive speeds, showcasing our Thar based autonomous vehicle navigating at near drift speeds, progressing towards our endeavour of Level-5 autonomy. Our autonomous vehicle at Swaayatt Robots was tasked with avoidance of traffic cones on the road, placed in a zig-zag fashion, at aggressive speeds. The location of the marked cones was not known to the planner beforehand. The #autonomousdriving task, i.e., motion planning (time parametrized trajectory computation) and decision making, was made even more challenging by restricting the AI agents to not act on obstacles unless they are within 24m radius. Level-5 #autonomousvehicles should be able to react quickly to overtake, or to avoid, any sudden unforeseeable obstacle or pedestrian on the road to avoid fatalities -- a capability demonstrated by our novel motion planning and decision making algorithmic framework over here. Our previous demo showcased our Bolero based platform consistently keeping speeds beyond 45 KMPH for most part, slowing down to only 39 KMPH at one point. Given Thar has lesser body roll, our framework successfully kept speeds well above 47 KMPH (even at the points of avoidance of obstacles), with speeds reaching as high as 55 KMPH. A typical human driver would feel uncomfortable at speeds beyond 40 KMPH in such as scenario. The entire algorithmic framework with 5 classical (one #reinforcementlearning-) agents , runs at 800+ Hz on a regular i7 processor, single thread. This algorithmic framework is being further scaled up with end-to-end deep reinforcement learning, and will be showcased in the month of March. #deeplearning #machinelearning #motionplanning

Sanjeev Sharma

12,839 次观看 • 1 年前

Multi-Track Timeline Control for Text-Driven 3D Human Motion Generation paper page: Recent advances in generative modeling have led to promising progress on synthesizing 3D human motion from text, with methods that can generate character animations from short prompts and specified durations. However, using a single text prompt as input lacks the fine-grained control needed by animators, such as composing multiple actions and defining precise durations for parts of the motion. To address this, we introduce the new problem of timeline control for text-driven motion synthesis, which provides an intuitive, yet fine-grained, input interface for users. Instead of a single prompt, users can specify a multi-track timeline of multiple prompts organized in temporal intervals that may overlap. This enables specifying the exact timings of each action and composing multiple actions in sequence or at overlapping intervals. To generate composite animations from a multi-track timeline, we propose a new test-time denoising method. This method can be integrated with any pre-trained motion diffusion model to synthesize realistic motions that accurately reflect the timeline. At every step of denoising, our method processes each timeline interval (text prompt) individually, subsequently aggregating the predictions with consideration for the specific body parts engaged in each action. Experimental comparisons and ablations validate that our method produces realistic motions that respect the semantics and timing of given text prompts.

AK

126,585 次观看 • 2 年前

Type a sentence, get any sound - from talking cats to singing saxophones. Brilliant release by NVIDIA ✨ NVIDIA just unveiled Fugatto, a groundbreaking 2.5B parameter audio AI model that can generate and transform any combination of music, voices, and sounds using text prompts and audio inputs Fugatto could ultimately allow developers and creators to bring sounds to life simply by inputting text prompts, → The model demonstrates unique capabilities like creating hybrid sounds (trumpet barking), changing accents/emotions in voices, and allowing fine-grained control over sound transitions - trained on millions of audio samples using 32 NVIDIA H100 GPUs 👨‍🔧 Architecture Built as a foundational generative transformer model leveraging NVIDIA's previous work in speech modeling and audio understanding. The training process involved creating a specialized blended dataset containing millions of audio samples → ComposableART's Innovation in Audio Control Introduces a novel technique allowing combination of instructions that were only seen separately during training. Users can blend different audio attributes and control their intensity → Temporal Interpolation Capabilities Enables generation of evolving soundscapes with precise control over transitions. Can create dynamic audio sequences like rainstorms fading into birdsong at dawn → Processes both text and audio inputs flexibly, enabling tasks like removing instruments from songs or modifying specific audio characteristics while preserving others → Shows capabilities beyond its training data, creating entirely new sound combinations through interaction between different trained abilities 🔍 Real-world Applications → Allows rapid prototyping of musical ideas, style experimentation, and real-time sound creation during studio sessions → Enables dynamic audio asset generation matching gameplay situations, reducing pre-recorded audio requirements → Can modify voice characteristics for language learning applications, allowing content delivery in familiar voices NVIDIA AI Developer

Rohan Paul

96,354 次观看 • 1 年前