| { |
| "checkpoint": { |
| "bytes": 160482975, |
| "config": { |
| "collision_max": 200.0, |
| "dropout": 0.1, |
| "fingerprint_dim": 2048, |
| "heads": 8, |
| "layers": 6, |
| "max_peaks": 256, |
| "metadata_scale": 0.02, |
| "model_dim": 384, |
| "mz_max": 2000.0, |
| "projection_dim": 192, |
| "retrieval_mlp_hidden_dim": 512, |
| "target_projection_dim": 256 |
| }, |
| "created_at": "2026-04-13T21:34:38Z", |
| "file": "weights/NexaMass-V3-Struct.pt", |
| "metrics_keys": [ |
| "backbone_lr", |
| "contrastive_loss", |
| "covariance_loss", |
| "covariance_to_contrastive_ratio", |
| "covariance_weight", |
| "dataloader_time_sec", |
| "dataloader_wait_pct_300s", |
| "dataloader_wait_pct_60s", |
| "dataloader_wait_pct_900s", |
| "dataloader_wait_pct_run", |
| "distributed_rank", |
| "distributed_world_size", |
| "effective_batch_size", |
| "effective_hard_negative_scale", |
| "effective_retrieval_weight", |
| "effective_structure_bce_weight", |
| "embedding_std", |
| "global_batch_size", |
| "gpu_active_fraction", |
| "gpu_count_observed", |
| "gpu_max_memory_gb", |
| "gpu_mem_used_mean_gb", |
| "gpu_mem_util_mean_pct", |
| "gpu_memory_utilization", |
| "gpu_snapshot_available", |
| "gpu_util_max_pct", |
| "gpu_util_mean_pct", |
| "gpu_util_min_pct", |
| "grad_norm", |
| "local_batch_size", |
| "lr", |
| "mfu_estimate", |
| "msms_objective", |
| "optimizer_step", |
| "regularizer_ramp", |
| "resumed_from_checkpoint", |
| "retrieval_head_lr", |
| "samples_per_sec", |
| "samples_per_sec_300s", |
| "samples_per_sec_60s", |
| "samples_per_sec_900s", |
| "samples_per_sec_run_mean", |
| "step_time_cv_300s", |
| "step_time_cv_60s", |
| "step_time_cv_900s", |
| "step_time_sec", |
| "structure_bce_locked", |
| "structure_bce_loss", |
| "structure_candidate_count", |
| "structure_clear_fraction", |
| "structure_clear_hit_at_1", |
| "structure_close_fraction", |
| "structure_close_hit_at_1", |
| "structure_confidence_bucket_score_mean", |
| "structure_confidence_bucket_score_mode", |
| "structure_confidence_candidate_count", |
| "structure_confidence_clear_threshold", |
| "structure_confidence_close_threshold", |
| "structure_confidence_eval_key_mode", |
| "structure_confidence_separation", |
| "structure_correct_logit_mean", |
| "structure_fingerprint_cosine", |
| "structure_hard_negative_bonus_applied", |
| "structure_hard_negative_bonus_mean", |
| "structure_hard_negative_scale", |
| "structure_head_lr", |
| "structure_hit_at_1", |
| "structure_hit_at_20", |
| "structure_hit_at_5", |
| "structure_local_ce_loss", |
| "structure_local_ce_loss_weight", |
| "structure_local_rerank_loss", |
| "structure_local_rerank_loss_weight", |
| "structure_local_rerank_score_scale", |
| "structure_logit_margin_mean", |
| "structure_margin_loss", |
| "structure_margin_loss_weight", |
| "structure_missing_formula_fraction", |
| "structure_missing_inchikey_fraction", |
| "structure_pairwise_logistic_loss", |
| "structure_pairwise_logistic_loss_weight", |
| "structure_positive_count_mean", |
| "structure_post_label_candidate_count", |
| "structure_post_precursor_candidate_count", |
| "structure_prefilter_candidate_count", |
| "structure_retrieval_logit_std", |
| "structure_retrieval_loss", |
| "structure_retrieval_scorer", |
| "structure_retrieval_target_mode", |
| "structure_skipped_query_fraction", |
| "structure_soft_target_exact_mix", |
| "structure_soft_target_loss", |
| "structure_soft_target_loss_weight", |
| "structure_target_backend", |
| "structure_target_equivalent_count_mean", |
| "structure_target_identity_collision_count", |
| "structure_target_rank_mean", |
| "structure_to_ssl_ratio", |
| "structure_top1_top2_gap_mean", |
| "structure_top1_top2_gap_norm_mean", |
| "structure_top_negative_logit_mean", |
| "structure_uncertain_fraction", |
| "structure_uncertain_hit_at_1", |
| "structure_valid_query_fraction", |
| "train_loss", |
| "train_loss_ema", |
| "train_retrieval_bank_size", |
| "variance_loss", |
| "variance_weight", |
| "weighted_covariance_loss", |
| "weighted_regularizer_to_contrastive_ratio", |
| "weighted_ssl_loss", |
| "weighted_structure_bce_loss", |
| "weighted_structure_retrieval_loss", |
| "weighted_variance_loss" |
| ], |
| "model_state_bytes": 56474172, |
| "model_state_file": "weights/NexaMass-V3-Struct-model_state.pt", |
| "param_count": 14106690, |
| "state_tensors": 103, |
| "step": 91001 |
| }, |
| "dataset": "corrected labeled GeMS handoff surface with enforced molecular labels", |
| "evaluation_summary": { |
| "embedding_std": 0.0722, |
| "gallery_exact_fingerprint_decodes": 11, |
| "gallery_size": 20, |
| "massspecgym_test_hit20_adapter": 0.3505, |
| "validation_fingerprint_cosine": 0.4255, |
| "validation_structure_bce": 0.0653 |
| }, |
| "name": "NexaMass-V3-Struct alignment checkpoint", |
| "objective_weights": { |
| "retrieval_local_ce_loss_weight": 0.4, |
| "retrieval_soft_target_loss_weight": 0.3, |
| "retrieval_weight": 0.6, |
| "ssl_preservation_weight": 0.05, |
| "structure_bce_weight": 1.0 |
| }, |
| "parent_checkpoint": "weights/Final_V3-model_state.pt", |
| "phase": "RDKit Morgan fingerprint structure alignment" |
| } |
|
|
