Learnable Align Attention Implementation

Abierto
#294 1 comentario 1 reacción 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
4/5
Tiempo estimado
3-5 días
Aptitud para principiantes
35/100
Tipo de issue
Error
Claridad
Bastante claro
Estado de actividad
Estancado
Stack tecnológico
python, tensorflow

Línea de trabajo

Start by locating the Learnable Align Attention implementation and examining the tf.einsum call that computes affinity. Compare its tensor dimensions with the DeepFusion paper’s description of correlations across voxel cells and camera features. Determine which shape the implementation should produce; the issue has no tests or file paths to point to, and done means resolving the dimensionality question and verifying the implementation matches the intended attention computation.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

In the DeepFusion paper it was said that

For each query (i.e., voxel cell), we conduct inner product between the query and the keys to obtain the attention affinity matrix that contains 1 × N correlations between the voxel and all its corresponding N camera features.

So I think this should lead to V x N correlations for V voxel cells and if we consider batches BxVxN. However in the implementation affinity = tf.einsum('bnc,bnc->bn', q, k) produces BxN shaped tensor. I feel like this should be affinity = tf.einsum("bij,bkl->bik",q,k). I couldnt manage to wrap my head around this, what am I missing?

Finally, thanks to the team for this great work.
@LiYingwei

Lenguaje dominante
Python
Estrellas
2.9k
Forks
452
Métricas de merge de PR
Sin PR fusionados en 30 d

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de tensorflow/lingvo

Todos los issues de tensorflow/lingvo

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.