Graph Attention Network (GAT) – Cơ Chế Chú Ý Trên Đồ Thị
Graph Attention Network (GAT) cải tiến GCN bằng cách cho phép mỗi cạnh học một trọng số chú ý khác nhau.
Tiến độ của tôi ở bài này
Điểm được lưu vào tài khoản sau khi chấm bài.
Đang tải điểm của bạn…
Kiến thức và chủ đề
Kiến thức tiên quyết: gin-layer-isomorphism, self-attention-mechanism.
Nội dung đề bài
Mục tiêu kiến thức
Graph Attention Network (GAT) cải tiến GCN bằng cách cho phép mỗi cạnh học một trọng số chú ý khác nhau.
Cơ chế chú ý cho cặp nút (i, j): eij = LeakyReLU(aT [Whi | Whj]) αij = softmaxj(eij) hi' = σ(∑j ∈ N(i) αij W hj)
Multi-head: concatenate (hoặc average) output của K heads.
Yêu cầu
Viết class:
class GATLayer:
def __init__(self, in_features, out_features, num_heads=1, concat=True, negative_slope=0.2):
pass
def forward(self, x, edge_index):
# x: [N, in_features], edge_index: [2, E]
# Returns: [N, out_features*num_heads] if concat else [N, out_features]
passInput
- Lớp
GATLayer(in_features,out_features,num_heads,concat,negative_slope): Khởi tạo đối tượng với các tham số, trọng số hoặc cấu hình tương ứng.
Output
- Các phương thức của
GATLayer: Trả về kết quả tính toán hoặc cập nhật trạng thái nội bộ của đối tượng.
Ràng buộc
- Thời gian chạy tối đa: 4000ms.
- Giới hạn bộ nhớ: 512MB.
- Dữ liệu đầu vào hợp lệ theo đúng kiểu dữ liệu và miền giá trị được mô tả.
Ví dụ 1
Input
gat = GATLayer(4, 8, num_heads=1, concat=True)
x = torch.randn(5, 4)
edge_index = torch.tensor([[0, 1, 2, 3], [1, 2, 3, 4]])
gat(x, edge_index)Output
Tensor shape: (5, 8), dtype=torch.float32Giải thích
Hàm/lớp được gọi với các tham số mẫu trên và trả về kết quả số học / kích thước tensor tương ứng theo đúng thiết kế.
Ví dụ 2
Input
gat = GATLayer(4, 8, num_heads=3, concat=True)
x = torch.randn(5, 4)
edge_index = torch.tensor([[0, 1, 2, 3], [1, 2, 3, 4]])
gat(x, edge_index)Output
Tensor shape: (5, 24), dtype=torch.float32Giải thích
Hàm/lớp được gọi với các tham số mẫu trên và trả về kết quả số học / kích thước tensor tương ứng theo đúng thiết kế.
Gợi ý và lời giải chỉ mở sau khi bạn bấm Nộp bài. Giáo viên và quản trị viên mở được ngay.
Góp ý & báo lỗi bài tập
Đề bài chưa rõ, test có vấn đề hay bạn có ý tưởng giúp bài tốt hơn? Gửi cho đội ngũ AI Empire nhé — mỗi góp ý đều được đọc.