kfac icon indicating copy to clipboard operation
kfac copied to clipboard

Is reusing linear layers the same as a convolutional layer

Open xmax1 opened this issue 5 years ago • 1 comments

If we are reusing weights in a linear layer can we use the same approximation to compute the covariances, or are there some subtleties?

for example if weights w are used 4x we can compute \Omega as (1/(4M)) A A^T where M is the batch size

deriving from the definition of a fisher block and assuming spatially uncorrelated derivatives seems to land you in the same place as the convolutional approximation

xmax1 avatar Jun 20 '20 01:06 xmax1

The type of approximation they use is the same one, essentially. However, you can't use them interchangeably in the code since convs involve special ops and also "reuse" the different patches of their input vectors.

james-martens avatar Jul 02 '20 15:07 james-martens