DiffCollage: Parallel Generation of Large Content with Diffusion Models
About
We present DiffCollage, a compositional diffusion model that can generate large content by leveraging diffusion models trained on generating pieces of the large content. Our approach is based on a factor graph representation where each factor node represents a portion of the content and a variable node represents their overlap. This representation allows us to aggregate intermediate outputs from diffusion models defined on individual nodes to generate content of arbitrary size and shape in parallel without resorting to an autoregressive generation procedure. We apply DiffCollage to various tasks, including infinite image generation, panorama image generation, and long-duration text-guided motion generation. Extensive experimental results with a comparison to strong autoregressive baselines verify the effectiveness of our approach.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Human Motion Composition | BABEL | PJ0.19 | 13 | |
| Multi-specific antibody co-design | Split-B | RMSD (Ag. A)3.27 | 7 | |
| Robot Manipulation | Compositional Planning Bench Tool-Use IND | Success Rate1 | 7 | |
| Robot Manipulation | Compositional Planning Bench Tool-Use OOD | Success Rate0.00e+0 | 7 | |
| Robot Manipulation | Compositional Planning Bench Drawer, IND | Success Rate0.00e+0 | 7 | |
| Robot Manipulation | Compositional Planning Bench Drawer OOD | Success Rate0.00e+0 | 7 | |
| Robot Manipulation | Compositional Planning Bench Cube, IND | Success Rate0.00e+0 | 7 | |
| Robot Manipulation | Compositional Planning Bench Cube, OOD | Success Rate0.00e+0 | 7 | |
| Robot Manipulation | Compositional Planning Bench Puzzle IND | Success Rate0.00e+0 | 7 | |
| Robot Manipulation | Compositional Planning Bench Puzzle OOD | Success Rate0.00e+0 | 7 |