Given some total system $ H = H_A \otimes H_B$, we can find the entanglement entropy for a subspace by taking $S = -tr(\rho_A ln \rho_A)$ with $\rho_A = tr_B \rho$, where $\rho$ is the density matrix of the system. For a two particles in a two level system (i.e. a state can be in $|0 \rangle$ or $|1 \rangle$, this seems quite straightforward. For example, for the state:
$$ \frac{1}{\sqrt{2}}(|00\rangle + |11 \rangle) $$
I could look at $\rho_A = \frac{1}{2}(|0\rangle_A \langle 0|_A+|1\rangle_A \langle 1|_A) $ and calculate the entanglement entropy from here.
I'm a little lost on how I would calculate entanglement entropy for identical particles in Fock space. For example, in the 3-Site Bose-Hubbard model with 3 particles, I can choose to label my states as $|n_1, n_2, n_3 \rangle$ where $n_i$ indicates the number of particles on site $i$. We thus have a 10 dimensional space: $|300\rangle, |030\rangle, |003 \rangle, ...$
I'm assuming that I can break down this system into three subspaces, each of just one particle. But how would I calculate the entanglement entropy from here?
Really my confusion comes from taking the partial traces in order to return a density matrix of just one particle. My assumption is that it can go something like this:
If our state is $|300\rangle$, I can rewrite this as $|1 \rangle \otimes |1\rangle \otimes | 1\rangle$, where now the number within the ket indicates what site the electron is on. Then, I can take the partial traces over two of the subspaces to return some $\rho_{A} = tr_Btr_c\rho$. This basically turns the 3-site Bose-Hubbard model into three, three level systems.
Is this approach correct?