Hi friends,

I can't reproduce the results of the MSNEA model. Could you share the code that replicates the results of the MSNEA model?

Sure, we have used the source code of MSNEA, modified part of it to make it fit our settings, which involves not using the content of attribute values but only the attribute types themselves.

We only made simple modifications, but this may affect the original performance of the model. If you have a better way, we welcome your feedback. In addition, we would be glad to release the full code for all baseline comparisons later on, which is currently being organized.

The main changes are shown below :

class AttrEncoder(nn.Module):
    def __init__(self, kgs, args):
        super().__init__()
        self.args = args
        # Using the attribute multi-hot data for attr_embed initialization
        self.attr_embed = nn.Embedding.from_pretrained(torch.FloatTensor(kgs["att_features"]))
        self.fc1 = nn.Linear(kgs["att_features"].shape[1], self.args.dim)
        self.fc2 = nn.Linear(self.args.dim, self.args.dim)
        nn.init.xavier_normal_(self.fc1.weight.data)
        nn.init.xavier_normal_(self.fc2.weight.data)

    def forward(self, e_idx, e_i):
        e_a = self.fc1(self.attr_embed(e_idx))
        # e_v = torch.sigmoid(e_v.unsqueeze(-1)).repeat(1, 1, self.args.dim)
        # e = self.fc2(torch.cat([e_a, e_v], dim=2))
        # Vision-adaptive Attribute Learning
        # alpha = F.softmax(torch.sum(e_a * e_i.unsqueeze(1), dim=-1), dim=1)
        # e = torch.sum(alpha.unsqueeze(2) * e_a, dim=1)
        return e_a

Could you share the code that replicates the results of the MSNEA model? #3

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions