LiTo: Surface Light Field Tokenization
Organizations: Apple
Abstract
We propose a 3D latent representation that jointly models object geometry and view-dependent appearance. Most prior works focus on either reconstructing 3D geometry or predicting view-independent diffuse appearance, and thus struggle to capture realistic view-dependent effects. Our approach leverages that RGB-depth images provide samples of a surface light field. By encoding random subsamples of this surface light field into a compact set of latent vectors, our model learns to represent both geometry and appearance within a unified 3D latent space. This representation reproduces view-dependent effects such as specular highlights and Fresnel reflections under complex lighting. We further train a latent flow matching model on this representation to learn its distribution conditioned on a single input image, enabling the generation of 3D objects with appearances consistent with the lighting and materials in the input. Experiments show that our approach achieves higher visual quality and better input fidelity than existing methods.
Figures & tables
| Method | Simple, Camera Radius [3, 4] | Hard, Camera Radius [1, 3] | ||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |
| TRELLIS | 31.12 3.39 | 0.974 0.022 | 0.034 0.022 | 27.57 3.38 | 0.941 0.050 | 0.090 0.055 |
| Ours | 34.16 3.39 | 0.985 0.016 | 0.023 0.018 | 32.36 3.77 | 0.967 0.040 | 0.055 0.046 |
| Method | Appearance | Latent size | PBR-Objaverse | Toys4k | GSO | |
| 0 | GT | – | – | 82.49 21.39 | 76.03 24.30 | 87.50 21.04 |
| Requires coarse geometry oracle: | ||||||
| 1 | TripoSF ( He et al., 2025 ) | ✗ | 244k 11 | 83.11 21.45 | 76.99 24.95 | 87.63 21.67 |
| 2 | TRELLIS ( Xiang et al., 2025 ) | ✓ | 20k 11 | 95.16 20.77 | 92.21 25.99 | 105.5 22.44 |
| 3-1 | 3DTopia-XL ( Chen et al., 2025b ) | ✓ | 2048 64 | 412.5 1129. | 153.9 305.8 | 98.50 19.66 |
| 3-2 | (worst 10% removed) | ✓ | 2048 64 | 135.4 95.75 | 90.61 23.96 | 93.62 12.86 |
| Method | CLIP | Conditioning View | Novel View | ||||||
| FID | KID | FID | KID | FID | KID | FID | KID | ||
| TRELLIS | 0.899 0.045 | 12.84 | 0.088 | 84.692 | 2.311 | 7.600 | 0.100 | 67.458 | 3.166 |
| Ours | 0.905 0.041 | 6.219 | 0.009 | 41.621 | 1.333 | 6.216 | 0.058 | 66.530 | 3.522 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | SH Deg | Enc Ray | Pred Occ | Mesh | Simple, Camera Radius [3, 4] | Hard, Camera Radius [1, 3] | CD (100k) | |||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |||||||
| Uniform Lighting | ||||||||||||
| 1-1 | TRELLIS | 0 | ✗ | - | ✓ | 28.17 4.09 | 0.970 0.024 | 0.039 0.024 | 24.63 4.01 | 0.934 0.054 | 0.098 0.059 | 95.19 26.41 |
| 1-2 | Ours | 0 | ✗ | ✗ | ✗ | 34.40 3.62 | 0.984 0.017 | 0.025 0.019 | 32.19 3.95 | 0.965 0.042 | 0.059 0.047 | 90.22 25.48 |
| 1-3 | Ours | 0 | ✓ | ✗ | ✗ | 34.44 3.47 | 0.984 0.017 | 0.026 0.020 | 32.18 3.79 | 0.964 0.042 | 0.060 0.048 | 89.24 25.34 |
| 1-4 | Ours | 1 | ✓ | ✗ | ✗ | 35.12 3.39 | 0.986 0.015 | 0.023 0.017 | 33.17 3.76 | 0.968 0.040 | 0.054 0.044 | 88.42 25.21 |
| Method | SH Deg | Enc Ray | Pred Occ | Mesh | Simple, Camera Radius [3, 4] | Hard, Camera Radius [1, 3] | CD (100k) | |||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |||||||
| Uniform Lighting | ||||||||||||
| 1-1 | TRELLIS | 0 | ✗ | - | ✓ | 27.34 3.82 | 0.947 0.036 | 0.053 0.029 | 23.72 3.66 | 0.883 0.068 | 0.139 0.065 | 108.2 22.56 |
| 1-2 | Ours | 0 | ✗ | ✗ | ✗ | 34.27 3.25 | 0.975 0.022 | 0.034 0.025 | 31.39 3.61 | 0.937 0.046 | 0.093 0.055 | 101.4 22.21 |
| 1-3 | Ours | 0 | ✓ | ✗ | ✗ | 34.04 3.23 | 0.974 0.022 | 0.034 0.025 | 31.15 3.59 | 0.935 0.048 | 0.093 0.055 | 100.0 21.41 |
| 1-4 | Ours | 1 | ✓ | ✗ | ✗ | 34.55 3.18 | 0.976 0.021 | 0.031 0.023 | 31.75 3.60 | 0.939 0.045 | 0.087 0.053 | 99.21 21.44 |
| Method | SH Deg | Enc Ray | Pred Occ | Mesh | Simple, Camera Radius [3, 4] | Hard, Camera Radius [1, 3] | CD (100k) | |||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |||||||
| Uniform Lighting | ||||||||||||
| 1-1 | TRELLIS | 0 | ✗ | - | ✓ | 28.63 3.09 | 0.955 0.028 | 0.046 0.025 | 25.06 2.93 | 0.902 0.057 | 0.121 0.062 | 98.09 22.21 |
| 1-2 | Ours | 0 | ✗ | ✗ | ✗ | 32.95 2.87 | 0.974 0.018 | 0.033 0.020 | 30.07 3.02 | 0.939 0.042 | 0.087 0.051 | 95.08 22.61 |
| 1-3 | Ours | 0 | ✓ | ✗ | ✗ | 33.14 2.68 | 0.974 0.018 | 0.034 0.021 | 30.21 2.85 | 0.937 0.042 | 0.089 0.053 | 94.16 22.55 |
| 1-4 | Ours | 1 | ✓ | ✗ | ✗ | 34.35 2.37 | 0.978 0.016 | 0.028 0.018 | 31.67 2.67 | 0.947 0.038 | 0.076 0.046 | 93.48 22.55 |
| Method | Simple, Camera Radius [3, 4] | Hard, Camera Radius [1, 3] | CD (100k) | |||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |||
| 150 input views | ||||||||
| 1 | TRELLIS | 31.559 3.509 | 0.9740 0.0224 | 0.0361 0.0217 | 27.948 3.539 | 0.9408 0.0508 | 0.0928 0.0539 | 90.65 25.13 |
| 2 | Ours | 33.909 3.157 | 0.9841 0.0162 | 0.0260 0.0189 | 32.073 3.521 | 0.9658 0.0403 | 0.0585 0.0458 | 80.54 27.62 |
| 120 input views | ||||||||
| 3 | TRELLIS | 31.518 3.509 | 0.9738 0.0225 | 0.0363 0.0218 | 27.912 3.541 | 0.9404 0.0510 | 0.0932 0.0541 | 90.88 25.19 |
| Method | Train w/ Ray | Infer w/ GT Ray | Train Iters | CLIP | Conditioning View | Novel View | |||||||
| FID | KID | FID | KID | FID | KID | FID | KID | ||||||
| 1 | TRELLIS | ✗ | - | 400k | 0.899 0.045 | 12.84 | 0.088 | 84.692 | 2.311 | 7.600 | 0.100 | 67.458 | 3.166 |
| 2-1 | Ours | ✗ | - | 280k | 0.906 0.040 | 8.193 | 0.012 | 48.117 | 0.461 | 6.648 | 0.064 | 75.814 | 4.321 |
| 2-2 | Ours | ✗ | - | 400k | 0.906 0.041 | 7.741 | 0.010 | 44.555 | 0.392 | 6.413 | 0.064 | 71.436 | 3.997 |
| 2-3 | Ours | ✗ | - | 600k | 0.905 0.041 | 6.219 | 0.009 | 41.621 | 1.333 | 6.216 | 0.058 | 66.530 | 3.522 |
| 3 | Ours | ✓ | ✗ | 290k | 0.900 0.040 | 10.78 | 0.066 | 65.644 | 2.281 | 8.076 | 0.101 | 92.915 | 6.698 |
| Occ Pred | Data Type | Method | Step | CLIP | Conditioning View | Novel View | |||||||
| FID | KID | FID | KID | FID | KID | FID | KID | ||||||
| 1 | ✗ | float32 | Heun | 100 | 0.905 0.041 | 6.219 | 0.009 | 41.621 | 1.333 | 6.216 | 0.058 | 66.530 | 3.522 |
| 2 | ✓ | float32 | Heun | 100 | 0.905 0.041 | 6.622 | 0.021 | 42.197 | 1.391 | 6.270 | 0.064 | 66.699 | 3.534 |
| 3 | ✓ | bfloat16 | Heun | 100 | 0.905 0.041 | 6.661 | 0.020 | 43.992 | 1.741 | 6.270 | 0.063 | 68.025 | 3.906 |
| 4 | ✓ | bfloat16 | Heun | 50 | 0.905 0.041 | 6.659 | 0.020 | 45.533 | 2.105 | 6.266 | 0.062 | 68.319 | 4.185 |
| 5 | ✓ | bfloat16 | Heun | 25 | 0.904 0.041 | 6.644 | 0.019 | 54.231 | 4.011 | 6.251 | 0.060 | 77.148 | 5.879 |
| Cond Proc (ms) | Structure Gen (s) | Latent Gen (s) | Occ Pred (ms) | 3DGS Dec (ms) | Mesh Dec (ms) | Total (s) | Memory (GB) | |
| NVIDIA A100-SXM4-80GB | ||||||||
| TRELLIS | 68.90 | 4.89 | 7.720 | – | 18.70 | 67.33 | 12.76 | 12.70 |
| Ours | 68.78 | – | 17.32 | 36.07 | 35.32 | 90.78 | 17.55 | 15.95 |
| NVIDIA H100 80GB HBM3 | ||||||||
| TRELLIS | 31.01 | 3.95 | 7.868 | – | 15.03 | 46.81 | 11.91 | 12.69 |
| Ours | 22.58 | – | 9.266 | 27.16 | 30.96 | 79.15 | 9.426 | 15.93 |
3D Representation Learning
3D representation learning builds geometric features for recognizing, reconstructing, generating, and reasoning about objects and scenes.
See all 187 papers